torch
torch里面包含所有的工具包已经tensor库
torch.nn
里面包含构造神经网络的一些基本模块
torch.autograd
里面包含一些对张量的操作
torch.nn.functional
里面包含一些神经网络常用的函数,比如损失函数,激活函数,卷积操作等等。
torch.optim
里面包含一些优化函数,比如的SGD(随机梯度下降)、Adam
torch.utils
里面包含一些对数据集的处理方法
torchvision
可以直接使用一些流行的数据集、模型等
关注我们,我们会每天坚持推送学到的新知识给你