

Distance functions


Loss functions


Vision functions


DataParallel functions (multi-GPU, distributed)


torch.Tensor


Data types


Initializing and basic operations


Tensor class reference


Tensor Attributes


torch.dtype


torch.device


torch.layout


torch.memory_format


Tensor Views


torch.amp


Autocasting


Gradient Scaling


Autocast Op Reference


torch.autograd


torch.autograd.backward


torch.autograd.grad


Forward-mode Automatic Differentiation


Functional higher level API


Locally disabling gradient computation


Default gradient layouts


In-place operations on Tensors


Variable (deprecated)


Tensor autograd functions


Function


Context method mixins


Numerical gradient checking


Profiler


Anomaly detection


Autograd graph


torch.library


torch.cuda


StreamContext


torch.cuda.can_device_access_peer


torch.cuda.current_blas_handle


torch.cuda.current_device

