Module phi.torch.nets

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

For API documentation, see https://tum-pbs.github.io/PhiFlow/Network_API .

Expand source code
"""
PyTorch implementation of the unified machine learning API.
Equivalent functions also exist for the other frameworks.

For API documentation, see https://tum-pbs.github.io/PhiFlow/Network_API .
"""
from phiml.backend.torch.nets import *
from phiml.backend.torch.nets import mlp as dense_net
from phiml.nn import parameter_count