pytorch中一些函数的用法

一.聚合函数

聚合函数指的是max/min，mean(求均值),sum等

以max函数为例，参数有（dim,keepdim=True/False）

注意：dim表示沿着哪个轴进行，keepdim表示操作之前的向量是几维的，操作之后维度保持不变

import torch
a=torch.arange(1,13).reshape(3,4)
print(a.max())
print(a.max(dim=0))
print(a.max(dim=0,keepdim=True))

输出的结果：注意这里max/min函数输出了索引

 二.查值操作

我们上面通过max函数得到索引，当然也可以引用argmax函数得到最大值的位置