Pytorch常用函数
一、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