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