常见的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()函数