max(input, dim, keepdim=False)
dim = 0按列为单位，返回该列的最大值及行索引
dim = 1 按行为单位，返回该行的最大值及列索引
未指定dim时则为整个数据集，只返回最大值
