device 查看tensor_Tensor数据操作

* torch.Tensor 存储和变换数据的工具

* Tensor与Ndarray多维数组类似，可提供GPU计算和自动求梯度

* 标量：0纬张量，一个数字

* 向量：1纬张量，[ ]

* 矩阵：2纬张量，[ [ ] ]

首先导入torch，查看版本号

importtorch#打印torch版本

print(torch.__version__)#查看是否支持GPUtorch.cuda.is_avaliable()

代码运行后的结果是：1.4.0

True

1.创建Tensor

1.11 直接创建

* torch.tensor(value)

* value:可以是任何维度的张量

print("创建一个标量5")

x= torch.tensor(5)print("创建一个向量[]")

y= torch.tensor([2,3])print("创建一个矩阵[ [ ] ]")

z= torch.tensor([[1,2],

[2,3]])print(x)print(y)print(z)

输出结果是：

创建一个标量5

创建一个向量[]

创建一个矩阵[ [ ] ]

tensor(5)

tensor([2, 3])

tensor([[1, 2],

[2, 3]])

1.2使用方法创建

* torch.empty(shape,dtype = type):创建未初始化的Tensor,可指定数据类型，dtype参数可省略

* torch.rand(shape) :创建随机矩阵

* torch.zeros(shape):创建0矩阵

* 创建维度均为2纬矩阵

print("创建二维未初始化矩阵")

A= torch.empty((2,2))print(A)print("创建随机矩阵")

B= torch.rand((2,2))print(B)print("创建0矩阵")

C= torch.zeros((2,2))print(C)

输出结果是：

创建二维未初始化矩阵

tensor([[0.0000e+00, 1.4279e+00],

[-9.3980e-11, 4.5908e-41]])