Pytorch 组成介绍

先知道以下 Torch 创始人大佬 Soumith Chintala

主要组件

  • *torch 张量库 *
  • torch.nn 包含用来构建神经网络的类
  • *torch.autograd 张量微分操作 *
  • torch.nn.functional 例如损失函数,激活函数
  • torch.optim 包含标准优化操作,如 SGD 和 ADam
  • torch.utils 数据集和数据加载器
  • torchvision 计算机视觉相关

转载请注明来源,欢迎对文章中的引用来源进行考证,欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论。
我的空间