目录
torch的基本功能:
①torch:张量的相关运算,例如:创建、索引、切片、连续、转置、加减乘除等相关运算。
②torch.nn:包含搭建网络层的模块(modules)和一系列的loss函数。
例如:全连接、卷积、池化、 BN分批处理、dropout、CrossEntropyLoss、MSLoss等。
包含五大基本功能模块:nn.Parameter、nn.Linear、nn.functional、nn.Module、nn.Sequential
③torch.autograd:提供Tensor所有操作的自动求导方法。
④torch.nn.functional:常用的激活函数relu、leaky_relu、sigmoid等。
⑤torch.optim:各种参数优化方法,例如:SGD、AdaGrad、RMSProp、Adam等。
⑥torch.nn.init:可以用它更改nn.Module的默认参数初始化方式。
⑦torch.utils.data:用于加载数据。
torch.nn库是专门为神经网络设计的模块化接口,nn构建于autograd之上,nn是Neural Network的简称,可以用来定义和运行神经网络,帮助程序员执行下述与神经网络相关的行为。
(1)创建神经网络
(2)训练神经网络
(3)保存神经网络
(4)恢复神经网络
其中torch.nn相关库的导入如下所示:
- import numpy as np # numpy数组库
- import math # 数学运算库
- import matplotlib.pyplot as plt # 画图库
-
- import torch # torch基础库
- import torch.nn as nn # torch神经网络库
- import torch.nn.functional as F
pip install torch===1.7.1 torchvision===0.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
出现如下界面即可:
注意:[Pytorch系列-30]:神经网络基础 - torch.nn库五大基本功能:nn.Parameter、nn.Linear、nn.functioinal、nn.Module、nn.Sequentia
(1)win+R输入cmd;
(2)然后切换到anaconda安装的位置,然后进入Script文件夹的位置。
(3)依次输入下述指令:
- pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
- pip install opencv-contrib-python -i https://pypi.tuna.tsinghua.edu.cn/simple
(4)输入conda list检查是否下载成功,出现下列标识即可。
(1)安装Numpy:因为Scipy依赖于Numpy,所以需要先安装Numpy,命令如下:
pip install numpy
(2)安装Scipy
pip install scipy