OneFlow Python API

oneflow
oneflow.nn
oneflow.nn.functional
oneflow.Tensor
Tensor Attributes
Type Info
oneflow.autograd
oneflow.cuda
oneflow.distributed
oneflow.distributions
- oneflow.distributions.Distribution
- oneflow.distributions.Categorical
oneflow.hub
oneflow.linalg
oneflow.nn.init
oneflow.optim
oneflow.nn.Graph
Auto Parallelism
oneflow.nn.image
oneflow.utils.data
oneflow.utils.global_view
oneflow.utils.tensor
- oneflow.utils.tensor.from_torch
- oneflow.utils.tensor.to_torch
oneflow.one_embedding
Environment Variables
oneflow.special

oneflow.nn.Module.forward¶