Pytorch常见包
2021-08-21 本文已影响0人
深思海数_willschang
- torch：包含所有的工具包以及tensor库
- torch.nn：包含构造神经网络的一些基本模块
- torch.nn.functional：包含一些神经网络常用的函数，比如损失函数，激活函数，卷积操作等等
- torch.optim：包含一些优化函数，比如的SGD（随机梯度下降)、Adam
- torch.autograd
- torch.utils：包含一些对数据集的处理方法
- torchvision：一些流行的数据集、模型等
