pytorch基本语法

Posted 2023-03-20

参考技术A pytorch类似于numpy的使用，同时可以使用GPU提升计算速度，还可以搭建深度学习模型，方便实用，简单易学，下面介绍一下pytorch的基本语法：

1.创建矩阵的方法

（1）构建未初始化的矩阵

（2）构建随机初始化矩阵

（3）构建全部为0，类型为long的矩阵

或者

（4）从数据直接直接构建tensor

（5）从一个已有的tensor构建一个tensor，new_ones重用以前的信息，例如数据类型；randn_like产生一个形状一样的矩阵。

2.显示数据类型，得到tensor的形状和改变形状

（1）显示数据类型

（2）得到tensor的形状

或者

（3）改变tensor形状

3.tensor运算

（1）加法

4.得到Python数值

5.Numpy，Tensor的CPU和GPU之间的转化

（1）把Torch Tensor转变成NumPy Array

（2）把NumPy ndarray转成Torch Tensor

（3）CPU转成GPU

（4）GPU不能直接转化成numpy，必须先转化为CPU,再转化为numpy