PyTorch
master (0.3.0+debian-1+Debian.stretch.9.3 )
Notes
Autograd mechanics
Broadcasting semantics
CUDA semantics
Extending PyTorch
Multiprocessing best practices
Serialization semantics
Package Reference
torch
torch.Tensor
torch.sparse
torch.Storage
torch.nn
torch.nn.functional
torch.nn.init
torch.optim
torch.autograd
torch.distributions
torch.multiprocessing
torch.distributed
torch.legacy
torch.cuda
Random Number Generator
Communication collectives
Streams and events
Memory management
NVIDIA Tools Extension (NVTX)
torch.utils.ffi
torch.utils.data
torch.utils.model_zoo
torch.onnx
PyTorch
Docs
»
torch.cuda
View page source
torch.cuda
¶
Random Number Generator
¶
Communication collectives
¶
Streams and events
¶
Memory management
¶
NVIDIA Tools Extension (NVTX)
¶