欢迎您访问 最编程 本站为您分享编程语言代码,编程技术文章!
您现在的位置是: 首页

详解Python中torch.amax函数的使用实例

最编程 2024-08-01 22:33:11
...

返回给定维度 diminput 张量的每个切片的最大值。

注意

max/minamax/amin之间的区别是:
  • amax /amin 支持多维缩减,

  • amax /amin 不返回索引,

  • amax /amin 在相等的值之间均匀分布梯度,而 max(dim) /min(dim) 仅将梯度传播到源张量中的单个索引。

如果 keepdimTrue ,则输出张量的大小与 input 相同,但在维度 dim 中它的大小为 1。否则,dim 被压缩(参见 torch.squeeze() ),导致输出张量的维度少 1 个(或 len(dim) )。

例子:

>>> 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])

推荐阅读