TypeError: max() received an invalid combination of arguments - got (axis=int, out=NoneType, ), but expected one of:

TypeError: max() received an invalid combination of arguments - got (axis=int, out=NoneType, ), but expected one of:
 * ()
 * (Tensor other)
 * (int dim, bool keepdim)
      didn't match because some of the keywords were incorrect: axis, out
 * (name dim, bool keepdim)
      didn't match because some of the keywords were incorrect: axis, out

 使用max函数出现报错

报错语句为

np.max(M, axis=0).reshape((-1, 1))

M=

tensor([[3.5500, 2.8013, 1.1312],
        [2.8497, 2.3735, 1.3523],
        [1.3359, 0.7867, 0.0713]])
改为

np.max(M.numpy(), axis=0).reshape((-1, 1))