PyTorch 中文文档
深度学习实践
刘帅(整理)
https://gitee.com/bleedingfight
这本书是关于学习 PyTorch 的使用和相关的 API、包的常见信息。资料均来自网络,为此本
书仅仅作为我的个人学习笔记,可以分享传播,主要内容来自PyTorch 中文。
First release, January 2019
Part I
PyTorch 中文
本书主要分为两部分:
• PyTorch 中文部分:这部分介绍了 PyTorch 常用的 API,主要包括如下模块:
– torch:torch 常用 API。
– torchvision:torchvision 常用 API
• PyTorch 使用示例:主要介绍 PyTorch 使用相关。
Contents
I PyTorch 中文
3
torch . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 11
11
12
19
27
31
32
32
1
1.1
Tensors
1.2 创建操作
1.3 索引, 切片, 连接, 换位
1.4 随机抽样
1.5 序列化
1.6 并行化
1.7.1 按点操作 ........................................................ 32
1.7 数学操作
1.8 累积操作
1.9 其它操作
1.10 BLAS/LAPACK 操作
1.11 Tensor
1.12
Storage
59
82
89
103
119
nn
1.13
1.13.1 Containers(容器) ............................................... 121
1.13.2 ParameterList.................................................... 128
1.13.3 卷积层......................................................... 129
1.13.4 池化层......................................................... 137
121
1.14 规范化层 (Normalization layers)
1.15 循环层
1.16 线性层
1.17 Dropout 层
1.18 稀疏层
1.19 损失函数
157
160
166
167
169
171
1.21
optim
179
181
1.20 Vision layers
1.22 Multi-GPU 层
1.21.1 如何使用optimizer................................................ 181
1.21.2 构建 .......................................................... 181
1.21.3 为每个参数单独设置选项 ........................................... 182
1.21.4 进行单次优化.................................................... 182
1.21.5 算法 .......................................................... 183
1.22.1 utils........................................................... 189
1.22.2 functional....................................................... 192
1.22.3 Pooling 函数..................................................... 194
1.22.4 Visionfunctions .................................................. 201
1.23.1 Variable........................................................ 202
1.23.2 init
........................................................... 207
1.24.1 Strategymanagement
.............................................. 211
1.24.2 SharingCUDAtensors .............................................. 211
1.24.3 Sharingstrategies ................................................. 212
1.24.4 Filedescriptor-ile_descripor ......................................... 212
1.24.5 Filesystem-ile_system ............................................. 212
1.24.6 fi
............................................................ 212
1.24 multiprocessing
autograd
1.23
189
202
211