- torch:包含所有的工具包以及tensor库
- torch.nn:包含构造神经网络的一些基本模块
- torch.nn.functional:包含一些神经网络常用的函数,比如损失函数,激活函数,卷积操作等等
- torch.optim:包含一些优化函数,比如的SGD(随机梯度下降)、Adam
- torch.autograd
- torch.utils:包含一些对数据集的处理方法
- torchvision:一些流行的数据集、模型等
本文标题:Pytorch常见包
本文链接:https://www.haomeiwen.com/subject/uasqfhtx.html
网友评论