PyTorch
master (0.2.0-1+Debian.stretch.9.1 )
Notes
Autograd mechanics
Broadcasting semantics
CUDA semantics
Extending PyTorch
Multiprocessing best practices
Serialization semantics
Package Reference
torch
Tensors
Creation Ops
Indexing, Slicing, Joining, Mutating Ops
Random sampling
Serialization
Parallelism
Math operations
Pointwise Ops
Reduction Ops
Comparison Ops
Other Operations
BLAS and LAPACK Operations
torch.Tensor
torch.sparse
torch.Storage
torch.nn
torch.nn.functional
torch.nn.init
torch.optim
torch.autograd
torch.multiprocessing
torch.distributed
torch.legacy
torch.cuda
torch.utils.ffi
torch.utils.data
torch.utils.model_zoo
torchvision Reference
torchvision
torchvision.datasets
torchvision.models
torchvision.transforms
torchvision.utils
PyTorch
Docs
»
torch
View page source
torch
¶
Tensors
¶
Creation Ops
¶
Indexing, Slicing, Joining, Mutating Ops
¶
Random sampling
¶
Serialization
¶
Parallelism
¶
Math operations
¶
Pointwise Ops
¶
Reduction Ops
¶
Comparison Ops
¶
Other Operations
¶
BLAS and LAPACK Operations
¶