PyTorch 中文文档
PyTorch 中文文档
Table of Contents
PyTorch 中文文档
- 翻译进度
PyTorch中文文档
自动求导机制
CUDA语义
- 最佳实践
扩展PyTorch
多进程最佳实践
- 共享CUDA张量
- 最佳实践和提示
序列化语义
- 最佳实践
张量 Tensors
数学操作Math operations
torch.Tensor
torch.Storage
torch.nn
torch.nn.functional
Automatic differentiation package - torch.autograd
- Variable
torch.optim
- 如何使用optimizer
- 算法
torch.nn.init
torch.multiprocessing
遗产包 - torch.legacy
torch.cuda
- 交流集
- 流和事件
torch.utils.ffi
torch.utils.data
torch.utils.model_zoo
torchvision
torchvision.datasets
- MNIST
- COCO
- LSUN
- ImageFolder
- Imagenet-12
- CIFAR
- STL10
torchvision.models
pytorch torchvision transform
torchvision.utils
- torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False)
- torchvision.utils.save_image(tensor, filename, nrow=8, padding=2, normalize=False, range=None, scale_each=False)
致谢
- 文档翻译

torch.nn

results matching ""