max

max(input, dim, keepdim=False)

dim = 0按列为单位，返回该列的最大值及行索引

dim = 1 按行为单位，返回该行的最大值及列索引

未指定dim时则为整个数据集，只返回最大值