Pytorch | Tensor張量
點擊上方“機器學習與生成對抗網(wǎng)絡”,關注星標
獲取有趣、好玩的前沿干貨!
01

02
import torch#torch.wherea = torch.rand(4, 4)b = torch.rand(4, 4)print(a)print(b)out = torch.where(a > 0.5, a, b)print(out)
print("torch.index_select")a = torch.rand(4, 4)print(a)out = torch.index_select(a, dim=0,index=torch.tensor([0, 3, 2]))#dim=0按列,index取的是行print(out, out.shape)

print("torch.gather")a = torch.linspace(1, 16, 16).view(4, 4)print(a)out = torch.gather(a, dim=0,index=torch.tensor([[0, 1, 1, 1],[],[]]))print(out)print(out.shape)

print("torch.masked_index")a = torch.linspace(1, 16, 16).view(4, 4)mask = torch.gt(a, 8)print(a)print(mask)out = torch.masked_select(a, mask)print(out)
print("torch.take")a = torch.linspace(1, 16, 16).view(4, 4)b = torch.take(a, index=torch.tensor([0, 15, 13, 10]))print(b)

#torch.nonzeroprint("torch.take")a = torch.tensor([[0, 1, 2, 0], [2, 3, 0, 1]])out = torch.nonzero(a)print(out)#稀疏表示

03
print("torch.stack")a = torch.linspace(1, 6, 6).view(2, 3)b = torch.linspace(7, 12, 6).view(2, 3)print(a, b)out = torch.stack((a, b), dim=2)print(out)print(out.shape)print(out[:, :, 0])print(out[:, :, 1])

04
05
import torcha = torch.rand(2, 3)print(a)out = torch.reshape(a, (3, 2))print(out)

print(a)print(torch.flip(a, dims=[2, 1]))print(a)print(a.shape)out = torch.rot90(a, -1, dims=[0, 2]) #順時針旋轉90°print(out)print(out.shape)
06
Tensor的填充操作
07

猜您喜歡:
CVPR 2021 | GAN的說話人驅(qū)動、3D人臉論文匯總
CVPR 2021生成對抗網(wǎng)絡GAN部分論文匯總
附下載 | 經(jīng)典《Think Python》中文版
附下載 |《TensorFlow 2.0 深度學習算法實戰(zhàn)》
評論
圖片
表情



