torch.amin¶
- torch.amin(input, dim, keepdim=False, *, out=None) Tensor ¶
返回给定维度(或维度)
dim
中input
张量每一片的最小值。注意
max
/min
和amax
/amin
之间的区别是amax
/amin
支持在多个维度上进行缩减,amax
/amin
不返回索引,amax
/amin
将梯度平均分配到相等的值之间,而max(dim)
/min(dim)
仅将梯度传播到源张量中的单个索引。
如果
keepdim
为True
,则输出张量的大小与input
相同,除了在维度dim
上,它的尺寸为 1。否则,dim
将被压缩 (参见torch.squeeze()
),导致输出张量减少 1 个 (或len(dim)
个) 维度。- 参数
- 关键字参数
out (Tensor, 可选) – 输出张量。
示例
>>> a = torch.randn(4, 4) >>> a tensor([[ 0.6451, -0.4866, 0.2987, -1.3312], [-0.5744, 1.2980, 1.8397, -0.2713], [ 0.9128, 0.9214, -1.7268, -0.2995], [ 0.9023, 0.4853, 0.9075, -1.6165]]) >>> torch.amin(a, 1) tensor([-1.3312, -0.5744, -1.7268, -1.6165])