output = torch.max(x,dim=1)
input输入的是一个tensor
dim是max函数索引的维度0/1，0是每列的最大值，1是每行的最大值
返回的是两个值：一个是每一行最大值的tensor组，另一个是最大值所在的位置
max_col_value = torch.max(x,dim=0)[0] # 每一列最大值
max_row_value = torch.max(x,dim=1)[0] # 每一行最大值
原文：https://www.cnblogs.com/douzujun/p/13427745.html