Module `phiml.backend.torch`

PyTorch integration.

Expand source code

"""
PyTorch integration.
"""
from ._torch_backend import TorchBackend as _TorchBackend

TORCH = _TorchBackend()
""" Backend for PyTorch operations. """

__all__ = [key for key in globals().keys() if not key.startswith('_')]

Sub-modules

phiml.backend.torch.nets: PyTorch implementation of the unified machine learning API. Equivalent functions also exist for the other frameworks …

Global variables

var TORCH: Backend for PyTorch operations.