torch.maximum¶ torch.maximum(input, other, *, out=None) → Tensor¶ 计算input和other的逐元素最大值。 注意 如果要比较的元素之一是 NaN,则返回该元素。对于具有复数 dtype 的张量,不支持maximum()。 参数 input (Tensor) – 输入张量。 other (Tensor) – 第二个输入张量 关键字参数 out (Tensor, 可选) – 输出张量。 示例 >>> a = torch.tensor((1, 2, -1)) >>> b = torch.tensor((3, 0, 4)) >>> torch.maximum(a, b) tensor([3, 2, 4])