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()函數


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM