Pytorch的max()与min()函数

参考 Pytorch的max()与min()函数 - 云+社区 - 腾讯云

一、max()函数

函数定义：torch.max(input, dim, max=None, max_indices=None，keepdim=False)

参数：

• input：进行max操作的Tensor变量
• dim：需要查找最大值得维度（这里很迷，后面重点介绍）
• max：结果张量，用于存储查找到的最大值
• max_indices：结果张量，用于存储查找到最大值所处的索引
• keepdim=False：返回值与原Tensor的size保持一致

1. 简单应用

t1=torch.LongTensor([3,9,6,2,5])

print("-------max-------")

print(torch.max(t1))

print("-------max dim-------")

print(torch.max(t1,dim=0))

输出结果为：