总结:
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