【torch.argmax与torch.max详解】

一、torch.max

1.调用方式

1）torch.max(input)：只需送入输入张量；
或
2）torch.max(input, dim, keepdim=False, *, out=None)：送入张量的同时，需要指定沿着哪个维度进行最大值运算；
这两种调用方式对输入张量的形状没有要求，一维数据或者多维数据都可以。

2.相关介绍

1）返回输入张量中最大值相关数据:

• 方式一，即不指定dim时，默认将张量展开成一维张量，然后返回第一个最大值；
• 方式二，即指定dim时，沿着指定的dim维进行最大值运算，输出结果由剩下的维度组成，比如原始维度为H,W，若指定dim=0（即H维),则输出结果由W个元素构成；

2）如果有多个最大值则返回第一个最大值；

3.代码实例及图示理解

首先定义一个简单的方法，当传入张量x和维度dim参数时，分别打印两种调用方式对应的输出：

def print_maxvalue(x