PyTorch 是基于以下两个目的而打造的python科学计算框架:
- 无缝替换NumPy,并且通过利用GPU的算力来实现神经网络的加速。
- 通过自动微分机制,来让神经网络的实现变得更加容易。
参考链接
- pytorch教程之nn.Module类详解——使用Module类来自定义模型,by LoveMIss-Y.
- PyTorch 中文教程,by apachecn.
- torch.distributions 详解,by Vic_Hao.
- What does log_prob do?,by stackoverflow.
- VAE中的重参数化技巧-reparameterization trick,by Weibo Mao.