PyTorch中文文档

PyTorch是使用GPU和CPU优化的深度学习张量库。

说明

• 自动求导机制
• CUDA语义
• 扩展PyTorch
• 多进程最佳实践
• 序列化语义

Package参考

• torch
• torch.Tensor
• torch.Storage
• torch.nn
• torch.nn.functional
• torch.nn.init
• torch.optim
• torch.autograd
• torch.multiprocessing
• torch.legacy
• torch.cuda
• torch.utils.ffi
• torch.utils.data
• torch.utils.model_zoo

torchvision参考

• torchvision
• torchvision.datasets
• torchvision.models
• torchvision.transforms
• torchvision.utils

致谢

• 致谢

艾伯特(http://www.aibbt.com/)国内第一家人工智能门户