Module phi.torch

PyTorch integration.

Importing this module registers the PyTorch backend with phiml.math. Without this, PyTorch tensors cannot be handled by phiml.math functions.

To make PyTorch the default backend, import phi.torch.flow.

Sub-modules

phi.torch.flow

Standard import for PyTorch mode …

phi.torch.nets

PyTorch implementation of the unified machine learning API. Equivalent functions also exist for the other frameworks …