Tensor的创建和维度的查看

常见的Tensor创建方法

1,基础Tensor函数:torch.Tensor(2,2)32位浮点型

2,指定类型: torch.DoubleTensor(2,2)64位浮点型

3,使用python的列表序列:torch.Tensor([[1,2],[3,4]])

4,默认值为0 :torch.zeros(2,3)

5,默认值为1 : torch.one(2,2)

6,对角张量 : torch.eye(2,2)

7,随机张量 :torch.randn(2,2)

8,随机排列张量  :torch.randperm(4)长度为4的随机排列向量

>>> a = torch.arange(1,5)
>>> a
tensor([1, 2, 3, 4])
>>> a = torch.arange(1,5,2)
>>> a
tensor([1, 3])

>>> a = torch.linspace(1,5,2)
>>> a
tensor([1., 5.])
>>> a = torch.linspace(1,5,3)
>>> a
tensor([1., 3., 5.])
#torch.tensor([1,2,3])

tensor([1,2,3])

维数的查看有两种  1,a.shape    2,a.size()

查看tensor中的元素的个数 a.numel()和a.nelement()函数

原文地址:https://www.cnblogs.com/kelvin-liu/p/14292850.html