OneFlow documentation

OneFlow documentation

Troubleshooting

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.cuda.current_device¶

oneflow.cuda.current_device() → int¶: Returns local rank as device index.

oneflow.cuda.set_device

oneflow.cuda.device_count

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