torch.max 和 torch.argmax的区别

qq_854297761

于 2021-10-11 10:28:24 发布

阅读量2.7k

点赞数 4

文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/qq_52643279/article/details/120697766

版权

本文详细介绍了PyTorch中torch.max和torch.argmax的区别。torch.max不仅返回最大值，还提供最大值的索引，而torch.argmax仅返回最大值的索引。此外，还简单提及了torch.min和torch.argmin的类似功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.torch.max

函数定义

torch.max(input, dim, max=None, max_indices=None, keepdim=False) -> (Tensor, LongTensor)

作用：找出给定tensor的指定维度dim上的上的最大值，并返回最大值在该维度上的值和位置索引。

应用举例：
例1——返回相应维度上的最大值,并返回最大值的位置索引

a = torch.randn(4, 4)
a
>tensor([[-1.2360, -0.2942, -0.1222,  0.8475],
            [ 1.1949, -1.1127, -2.2379, -0.6702],
            [ 1.5717, -0.9207,  0.1297, -1.8768],
            [-0.6172,  1.0036, -0.6060, -0.2432]])
torch.max(a, 1)
>torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), 
indices=tensor([3, 0, 0, 1]

最低0.47元/天解锁文章