美文网首页
Pytorch常见包

Pytorch常见包

作者: 深思海数_willschang | 来源:发表于2021-08-21 09:48 被阅读0次
  • torch:包含所有的工具包以及tensor库
  • torch.nn:包含构造神经网络的一些基本模块
  • torch.nn.functional:包含一些神经网络常用的函数,比如损失函数,激活函数,卷积操作等等
  • torch.optim:包含一些优化函数,比如的SGD(随机梯度下降)、Adam
  • torch.autograd
  • torch.utils:包含一些对数据集的处理方法
  • torchvision:一些流行的数据集、模型等

相关文章

网友评论

      本文标题:Pytorch常见包

      本文链接:https://www.haomeiwen.com/subject/uasqfhtx.html