max, argmax, maximum傻傻分不清楚

biubiu713

已于 2023-05-25 16:05:00 修改

阅读量186

点赞数

分类专栏： pytorch基础文章标签： python 深度学习 pytorch

于 2023-05-25 16:02:54 首次发布

本文链接：https://blog.csdn.net/biubiu713/article/details/130868129

版权

pytorch基础专栏收录该内容

7 篇文章 0 订阅

订阅专栏

该文介绍了PyTorch中用于找出张量最大值和对应索引的函数torch.max()和torch.argmax()，以及计算两个张量逐元素最大值的torch.maximum()。torch.max()可返回张量所有元素或指定维度的最大值及索引，而torch.argmax()提供最大值的索引。torch.maximum()则比较两个张量并返回较大值的结果张量，允许张量维度不同。

摘要由CSDN通过智能技术生成

torch.max()

torch.max(input: Tensor) --> Tensor: 返回所有元素的最大值
torch.max(input: Tensor, dim: int, keepdim: bool = False) --> Tuple(Tensor, Tensor): 返回指定维度的最大值和对应的索引
- 如果keepdim=True，维度不变，指定维度的大小为1
torch.Tensor.max() 参考 torch.max()

代码示例

import torch
x = torch.randn(3, 4)
max_value = x.max()
max_values, max_indices = x.max(dim=0)

torch.argmax()

torch.argmax(input: Tensor) --> Tensor：返回所有元素最大值的索引
torch.argmax(input: Tensor, dim: int, keepdim: bool = False)：返回指定维度的最大值对应的索引
torch.Tensor.argmax() 参考 torch.argmax()

代码示例

x = torch.randn(3, 4)
max_index = x.argmax()
max_indices = x.argmax(dim=0)

torch.maximum()

torch.maximum(input: Tensor, other: Tensor) --> Tensor: 返回input tensor 和 other tensor 的逐元素最大值
other的维度可以与input的维度不同，但维度较少的tensor的维度大小要与维度较多的tensor的最后维度大小保持一致
torch.Tensor.maximum() 参考 torch.maximum()
代码示例

# 维度和维度大小相同
x1 = torch.randn(2, 3, 4)
y1 = torch.randn(2, 3, 4)

# 维度不同，维度较少的tensor的维度大小与维度较多的tensor的最后维度大小保持一致
x2 = torch.randn(1, 2, 3, 4)
y2 = torch.randn(2, 3, 4)

z1 = torch.maximum(x1, y1)
z2 = torch.maximum(x2, y2)

print(z1.shape, z2.shape)
# torch.Size([2, 3, 4]) torch.Size([1, 2, 3, 4])