背景
0.1版本 THNN后端
1.0版本 CAFFE2后端
torch基础上开发
同类框架
Google:tensorflow()
facebook:pytorth(静态库)
静态图:公式固定
动态图:公式可变
PyTorch生态
自然语言处理包
AllenNLP
PyTorthNLP
视觉包
TorchVision
图卷积包
PyTorch geometric
Fast.ai
部署库的协议
PyTorch训练的模型通过ONNX部署在移动设备上
PyTchor能做什么
1 GPU加速
2 自动求导
3 常用网络层(调用函数层函数,搭积木一样搭建复杂的神经网络结构.一层层堆叠复杂的神经网络结构)
nn.Linear
nn.Conv2d
nn.LSTM
nn.ReLU
nn.Sigmoid
nn.Softmax
nn.CrossEntropyLoss
nn.MSE