Tensor的建立和維度的檢視

2022-07-10 08:15:12 字數 773 閱讀 1419

常見的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()函式

tensor的維度變換

expand與repeat用法 t 轉置 transpose轉置 permute函式 例子 a torch.rand 4,1,28,28 a.view 4,28 28 例子2 a torch.rand 4,1,28,28 a.view 4 1,28,28 表示我們現在只關注feature map這個...

建立和檢視索引

create table table name 屬性名1 資料型別,屬性名2 資料型別,屬性名3 資料型別,index index name 屬性1 例如 create table t dept deptno int,dname varchar 20 loc varchar 20 index ind...

PyTorch中張量 tensor 的維度變換

example in 1 x torch.rand 4,1,28,28 in 2 x.size out 2 torch.size 4,1,28,28 in 3 y x.view 4,28 28 in 4 y.size out 4 torch.size 4,784 in 5 y x.reshape 4...