OneFlow documentation

OneFlow documentation

Troubleshooting

OneFlow Python API

oneflow
oneflow.nn
oneflow.nn.functional
oneflow.Tensor
Tensor Attributes
oneflow.autograd
oneflow.cuda
oneflow.distributed
oneflow.linalg
oneflow.nn.init
oneflow.optim
oneflow.nn.Graph
oneflow.nn.image
oneflow.utils.data
oneflow.one_embedding
Environment Variables

oneflow.env.get_world_size¶

oneflow.env.get_world_size()¶

Returns the number of processes in the current process group.

Returns: The world size of the process group.

oneflow.env.get_rank

oneflow.distributed

Copyright © 2020, OneFlow | Built with Sphinx and @pradyunsg's Furo theme. | Show Source