天天看點

pytorch中 max()、view()、 squeeze()、 unsqueeze()

總結:

max():

  1. max(a),用于一維資料,求出最大值。
  2. max(a,0),計算出資料中一列的最大值,并輸出最大值所在的行号。
  3. max(a,1),計算出資料中一行的最大值,并輸出最大值所在的列号。

a.view(i,j)表示将原矩陣轉化為i行j列的形式  

  1. a.view(i,-1)表示不限制列數,輸出i行
  2. a.view(-1,j)表示不限制行數,輸出j列

a.squeeze(i)   壓縮第i維,如果這一維維數是1,則這一維可有可無,便可以壓縮

unsqueeze(i) 表示将第i維設定為1