Tensor

Tensor 基本数据类型

32位浮点数	torch.float32
64位浮点数	torch.float64
32位整型	torch.int.32
16位整型	torch.int.16
64位整型	torch.int.64

Tensor 与 numpy 转化

• torch.from_numpy(numpy矩阵)

• Tensor矩阵.numpy(）

例子：

x =torch.tensor([6,2],dtype=torch.float32)

x =torch.tensor([6,2],dtype=torch.float32)
x
Out[14]: tensor([6., 2.])
x =torch.zeros([6,2],dtype=torch.float32)
x
Out[16]: 
tensor([[0., 0.],
        [0., 0.],
        [0., 0.],
        [0., 0.],
        [0., 0.],
        [0., 0.]])