Python API Torch

torch.cuda = GPU-related operations torch.nn = Neural networks torch.optim = Optimization algorithms torch.utils.data = Data loading and processing

torch.fft = Fast Fourier Transform functions torch.distributed = Distributed computing torch.profiler = Performance profiling torch.special = Special mathematical functions

torch.autograd = Automatic differentiation torch.jit = Just-in-time compilation torch.linalg = Linear algebra operations torch.onnx = Open Neural Network Exchange format

<p>torch.nn = Neural networks torch.autograd = Automatic differentiation torch.cuda = CUDA operations torch.optim = Optimization algorithms</p> Signup and view all the answers

<p>torch.distributed = Distributed computing torch.fft = Fast Fourier Transform operations torch.jit = JIT compilation torch.linalg = Linear algebra operations</p> Signup and view all the answers

<p>torch.monitor = Monitoring tools torch.signal = Signal processing operations torch.package = Packaging utilities torch.profiler = Profiling tools</p> Signup and view all the answers

<p>torch.nn.init = Neural network initialization torch.onnx = ONNX model operations torch.random = Random number generation torch.masked = Masked tensor operations</p> Signup and view all the answers

<p>torch.nested = Nested tensor operations torch.sparse = Sparse tensor operations torch.Storage = Tensor storage operations torch.utils.benchmark = Benchmarking utilities</p> Signup and view all the answers

<p>torch.utils.bottleneck = Performance bottleneck detection torch.utils.checkpoint = Checkpointing utilities torch.utils.cpp_extension = C++ extension utilities torch.utils.data = Data loading and processing</p> Signup and view all the answers

<p>torch.utils.jit = JIT compilation utilities torch.utils.dlpack = DLPack interoperability torch.utils.mobile_optimizer = Mobile optimization torch.utils.model_zoo = Pretrained model repository</p> Signup and view all the answers

<p>torch.utils.tensorboard = TensorBoard utilities torch.<strong>config</strong> = Configuration operations torch.distributed.algorithms.join = Join-based distributed algorithms torch.distributed.elastic = Elastic distributed training</p> Signup and view all the answers