Pytorch库的基本架构介绍

今天我就将PyTorch常用的模块做一个总结梳理。

首先要说明的是PyTorch这是torch的Python版本，所以导入的是torch而不是Pytorch:

import torch

 

1 运行基础

• torch.tensor：基础数据结构

• torch.autograd：自动微分模块

2 torch.utils

支持神经网络相关的数据预处理。

• 数据导入与处理

  • utils.data

  • utils.datasets

• utils.tensorboard：训练结果的可视化

• utils.model_zoo：预训练模型

3 torch.nn

构建神经网络结构的基本元素。

• nn.Module：神经网络的各种结构“层”

• nn.functional:神经网络的损失函数与激活函数

4 torch.optim

神经网络的算法优化模块，封装着各类优化器。

5 神经网络的运算性能

文章来源: wenyusuran.blog.csdn.net，作者：文宇肃然，版权归原作者所有，如需转载，请联系作者。

原文链接：wenyusuran.blog.csdn.net/article/details/123637677