Python API

Created On: Apr 16, 2025 | Last Updated: Apr 16, 2025

• torch
• torch.nn
• torch.nn.functional
• torch.Tensor
• Tensor Attributes
• Tensor Views
• torch.amp
• torch.autograd
• torch.library
• torch.accelerator
• torch.cpu
• torch.cuda
• Understanding CUDA Memory Usage
• Generating a Snapshot
• Using the visualizer
• Snapshot API Reference
• torch.mps
• torch.xpu
• torch.mtia
• torch.mtia.memory
• Meta device
• torch.backends
• torch.export
• torch.distributed
• torch.distributed.tensor
• torch.distributed.algorithms.join
• torch.distributed.elastic
• torch.distributed.fsdp
• torch.distributed.fsdp.fully_shard
• torch.distributed.tensor.parallel
• torch.distributed.optim
• torch.distributed.pipelining
• torch.distributed.checkpoint
• torch.distributions
• torch.compiler
• torch.fft
• torch.func
• torch.futures
• torch.fx
• torch.fx.experimental
• torch.hub
• torch.jit
• torch.linalg
• torch.monitor
• torch.signal
• torch.special
• torch.overrides
• torch.package
• torch.profiler
• torch.nn.init
• torch.nn.attention
• torch.onnx
• torch.optim
• Complex Numbers
• DDP Communication Hooks
• Quantization
• Distributed RPC Framework
• torch.random
• torch.masked
• torch.nested
• torch.Size
• torch.sparse
• torch.Storage
• torch.testing
• torch.utils
• torch.utils.benchmark
• torch.utils.bottleneck
• torch.utils.checkpoint
• torch.utils.cpp_extension
• torch.utils.data
• torch.utils.deterministic
• torch.utils.jit
• torch.utils.dlpack
• torch.utils.mobile_optimizer
• torch.utils.model_zoo
• torch.utils.tensorboard
• torch.utils.module_tracker
• Type Info
• Named Tensors
• Named Tensors operator coverage
• torch.__config__
• torch.__future__
• torch._logging
• Torch Environment Variables