在之前的博客已经对Pytorch进行了介绍及安装,接下来正式学习Pytorch,本章博客主要介绍torch Tensor、torch Tensor与numpy array的转化以及运算操作。
1.Tensor
Tensor类似于numpy的ndarrays,另外它还可以在GPU上使用Tensor进行加速计算。
导入pytorch包
import torch
下面讲解如何创建一个5*3的矩阵,并输出
x = torch.Tensor(5, 3)
print(x)
#输出:
0.0000 0.0000 0.0000
0.0000 0.4677 0.0000
0.4692 0.0000 0.4662
0.0000 0.4693 0.0000
12.7686 189.2708 0.0000
[torch.FloatTensor of size 5x3]
创建一个随机初始化矩阵
x = torch.rand(5, 3)
print(x)
#输出:
0.0608 0.8530 0.3833
0.3624 0.2066 0.9663
0.6356