PyTorch 2.6.0版本

• 概述

• torch

• torch.nn

• torch.nn.functional

• torch.Tensor

• Tensor Attributes

• torch.amp

• torch.autograd

• torch.library

• torch.accelerator

• torch.cuda

• torch.backends

• torch.distributed

• torch.distributed.algorithms.join

• torch.distributed.elastic

• torch.distributed.fsdp

• torch.distributed.optim

• torch.distributed.pipelining

• torch.distributed.tensor.parallel

• torch.distributed.checkpoint

• torch.distributions

• torch.fft

• torch.func

• torch.futures

• torch.fx

• torch.hub

• torch.jit

• torch.linalg

• torch.monitor

• torch.special

• torch.overrides

• torch.package

• torch.profiler

• torch.nn.init

• torch.onnx

• torch.optim

• torch.cpu

• torch.compiler

• torch.export

• DDP Communication Hooks

• Quantization

• Distributed RPC Framework

• torch.random

• torch.sparse

• torch.Storage

• torch.testing

• torch.utils

• torch.utils.benchmark

• torch.utils.checkpoint

• torch.utils.cpp_extension

• torch.utils.data

• torch.utils.dlpack

• torch.utils.mobile_optimizer

• torch.utils.model_zoo

• torch.utils.tensorboard

• Type Info

• Named Tensors

• torch.__config__