PyTorch 文档¶
PyTorch 是一个优化的张量库，用于使用 GPU 和 CPU 进行深度学习。
本文档中描述的功能按版本状态分类：
稳定：这些功能将长期维护，通常应该有 在文档中没有重大的性能限制或差距。 我们还希望保持向后兼容性（尽管 可能会发生重大更改，并且将提前一个版本发出通知 的时间）。
试用版：这些功能被标记为 Beta 版，因为 API 可能会根据 用户反馈，因为性能需要改进，或者因为 跨 Operators 的覆盖范围尚未完成。对于 Beta 版功能，我们是 致力于将该功能一直持续到 Stable 分类。 但是，我们并不承诺向后兼容。
原型：这些功能通常不作为 PyPI 或 Conda 等二进制发行版，但有时落后于运行时 标志，并且处于反馈和测试的早期阶段。
开发人员说明
语言绑定
Python 接口
- torch
- torch.nn
- torch.nn.functional
- torch.Tensor
- 张量属性
- 张量视图
- torch.amp
- torch.autograd
- torch.library
- torch.cuda
- torch.backends
- torch.distributed
- torch.distributed.algorithms.join
- torch.distributed.elastic
- torch.distributed.fsdp
- torch.distributed.optim
- torch.distributions
- torch.fft
- 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
- 复数
- DDP 通信钩子
- 管道并行度
- 量化
- 分布式 RPC 框架
- torch.random
- torch.masked
- torch.nested
- torch.sparse
- torch.Storage
- torch.testing
- torch.utils.benchmark
- torch.utils.bottleneck
- torch.utils.checkpoint
- torch.utils.cpp_extension
- torch.utils.data
- torch.utils.jit
- torch.utils.dlpack
- torch.utils.mobile_optimizer
- torch.utils.model_zoo
- torch.utils.tensorboard
- 类型信息
- 命名张量
- 命名张量算子覆盖率
- torch.__config__