【冰糖Python】PyTorch：最大值 最小值 torch.max() torch.min() torch.maximum() torch.minimum()

获得 Tensor 元素中的最大值、最小值

1、torch.max()

torch.max(input, dim, keepdim=False)

返回命名元组（最大值，最大值索引），最大值是给定维度中的最大值，索引为在对应维度中的索引

当有多个最大值时，返回第一个最大值的索引

当没有指定维度 dim 时，则返回所有元素中的最大值（single value）

input：输入张量

dim：指定的维度

keepdim：默认 False，输出是否保留原始维度

torch 中还给出了直接获取最大值索引的方法 torch.argmax()

其用法和 torch.max() 一致，且结果即为 torch.max() 返回的第二个元组（索引的元组）

2、torch.min()

torch.min(input, dim, keepdim=False)

返回命名元组（最小值，最小值索引），最小值是给定维度中的最小值，索引为在对应维度中的索引

当有多个最小值时，返回第一个最小值的索引

当没有指定维度 di