torch.max()使用讲解

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]     # 每一行最大值