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