4.1.6.4 pde.backends.torch.typing module

Defines types specific to the torch backend.

class TorchInnerStepperType(*args, **kwargs)

Bases: Protocol

General backend-level stepping-function type working with torch tensors.

class TorchRHSType(*args, **kwargs)

Bases: Protocol

General right-hand-side function type working with torch tensors.