天天看點

Pytorch學習記錄(常用函數1)

img = np.array(Image.open(''cat.png''))  # 加載圖檔并轉為numpy格式

inp = torch.from_numpy(img)    将numpy格式資料轉為tensor

permute(2, 0, 1)  # 将三維張量的次元進行變換,原始為0,1,2,變換後為2,0,1,可以了解我軸的變換。

transpose(1,2)  # 将次元1 ,2進行調換,是以permute用于多個次元調換,transpose用于兩個調換

unsqueeze()     # 擴充次元

squeeze()   # 将長度為1的次元壓縮 

flatten()  # 将資料進行拼接,減少次元

expand()  # 将原始張量擴充為目标張量,參數為擴張後變量大小

torch.cat()  #

@ 矩陣相乘

* 逐元素相乘

繼續閱讀