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.Tensor.pin_memory¶

Tensor.pin_memory() → Tensor ¶: Copies the tensor to pinned memory, if it’s not already pinned.

oneflow.Tensor.is_pinned

oneflow.Tensor.nms

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