4.1.6.4 pde.backends.torch.typing module

Defines types specific to the torch backend.

class TorchInnerStepperType(*args, **kwargs)

Bases: Protocol

General stepper type working with torch tensors.

class TorchRHSType(*args, **kwargs)

Bases: Protocol

General stepper type working with torch tensors.