Module phi.torch
PyTorch integration.
Importing this module registers the PyTorch backend with phi.math
.
Without this, PyTorch tensors cannot be handled by phi.math
functions.
To make PyTorch the default backend, import phi.torch.flow
.
Expand source code
"""
PyTorch integration.
Importing this module registers the PyTorch backend with `phi.math`.
Without this, PyTorch tensors cannot be handled by `phi.math` functions.
To make PyTorch the default backend, import `phi.torch.flow`.
"""
from phi import math as _math
from ._torch_backend import TorchBackend as _TorchBackend
TORCH = _TorchBackend()
""" Backend for PyTorch operations. """
_math.backend.BACKENDS.append(TORCH)
__all__ = [key for key in globals().keys() if not key.startswith('_')]
Sub-modules
phi.torch.flow
-
Standard import for PyTorch mode …
phi.torch.nets
Global variables
var TORCH
-
Backend for PyTorch operations.