PyTorch从入门到精通100讲(四)-神经网络在pytorch中的应用
【摘要】
Pytorch库的基本架构介绍
今天我就将PyTorch常用的模块做一个总结梳理。
首先要说明的是PyTorch这是torch的Python版本,所以导入的是torch而不是Pytorch:
import torch
1 运行基础
torch.tensor:基础数据结构
torch.autograd:自...
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
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
评论(0)