天天看点

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