torch.amax¶
- torch.amax(input, dim, keepdim=False, *, out=None) Tensor ¶
返回给定维度(s)
dim
中input
张量每个切片的最大值。注意
max
/min
和amax
/amin
之间的区别是amax
/amin
支持在多个维度上进行缩减,amax
/amin
不返回索引,amax
/amin
在相等的值之间均匀分配梯度,而max(dim)
/min(dim)
仅将梯度传播到源张量中的单个索引。
如果
keepdim
为True
,则输出张量的大小与input
相同,除了维度(s)dim
的大小为 1。否则,dim
会被压缩(参见torch.squeeze()
),导致输出张量具有 1(或len(dim)
)个较少的维度。- 参数
- 关键字参数
out (Tensor, 可选) – 输出张量。
示例
>>> a = torch.randn(4, 4) >>> a tensor([[ 0.8177, 1.4878, -0.2491, 0.9130], [-0.7158, 1.1775, 2.0992, 0.4817], [-0.0053, 0.0164, -1.3738, -0.0507], [ 1.9700, 1.1106, -1.0318, -1.0816]]) >>> torch.amax(a, 1) tensor([1.4878, 2.0992, 0.0164, 1.9700])