PyTorch中的torch.max()和torch.maximum()的用法详解

Posted 2021-12-11 魏晓蕾

1、torch.max()

torch.max(input) → Tensor

返回 input tensor 中所有元素的最大值。

torch.max(input, dim, keepdim=False) → output tensors (max, max_indices)

参数：

• input：输入的 tensor。
• dim：按什么维度求最大值（2D中，0代表按列求最大值，1代表按行求最大值）。
• keepdim：是否保持 input tensor 的维度，True 代表 out tensor 与 input tensor 的维度相同，False 代表 out tensor 与 input tensor 的维度不同。

返回值：

• 返回一个 namedtuple (values, indices)，values表示指定维度的最大值，indices表示最大值所在的索引，如果给定维度有多个最大值，返回第一个最大值所在的索引。

2、torch.maximum()

torch.maximum(input, other) → Tensor

参数：

• input：第一个输入 tensor。
• other：第二个输入 tensor。

返回值：

• 两个 tensor 进行逐元素比较，返回每个较大的元素组成一个新的 tensor。如果元素中有 NaN 值，则返回 NaN 值。