總結:
max():
- max(a),用于一維資料,求出最大值。
- max(a,0),計算出資料中一列的最大值,并輸出最大值所在的行号。
- max(a,1),計算出資料中一行的最大值,并輸出最大值所在的列号。
a.view(i,j)表示将原矩陣轉化為i行j列的形式
- a.view(i,-1)表示不限制列數,輸出i行
- a.view(-1,j)表示不限制行數,輸出j列
a.squeeze(i) 壓縮第i維,如果這一維維數是1,則這一維可有可無,便可以壓縮
unsqueeze(i) 表示将第i維設定為1