Pytorch | Tensor張量
點(diǎn)擊上方“機(jī)器學(xué)習(xí)與生成對(duì)抗網(wǎng)絡(luò)”,關(guān)注星標(biāo)
獲取有趣、好玩的前沿干貨!
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]) #順時(shí)針旋轉(zhuǎn)90°print(out)print(out.shape)
06
Tensor的填充操作
07

猜您喜歡:
等你著陸!【GAN生成對(duì)抗網(wǎng)絡(luò)】知識(shí)星球!
CVPR 2021 | GAN的說(shuō)話人驅(qū)動(dòng)、3D人臉論文匯總
CVPR 2021 | 圖像轉(zhuǎn)換 今如何?幾篇GAN論文
【CVPR 2021】通過GAN提升人臉識(shí)別的遺留難題
CVPR 2021生成對(duì)抗網(wǎng)絡(luò)GAN部分論文匯總
最新最全20篇!基于 StyleGAN 改進(jìn)或應(yīng)用相關(guān)論文
附下載 | 經(jīng)典《Think Python》中文版
附下載 | 《Pytorch模型訓(xùn)練實(shí)用教程》
附下載 | 最新2020李沐《動(dòng)手學(xué)深度學(xué)習(xí)》
附下載 | 《可解釋的機(jī)器學(xué)習(xí)》中文版
附下載 |《TensorFlow 2.0 深度學(xué)習(xí)算法實(shí)戰(zhàn)》
評(píng)論
圖片
表情



