深入浅出Pytorch函数——torch.maximum

分类目录：《深入浅出Pytorch函数》总目录

相关文章：

· 深入浅出Pytorch函数------torch.max

· 深入浅出Pytorch函数------torch.maximum

---

计算input和other的元素最大值。

语法

torch.maximum(input, other, *, out=None) -> Tensor

参数

• input：[Tensor] 输入张量
• other：[Tensor] 输入的第二个张量

实例

>>> a = torch.randn(4, 5)
>>> b = torch.randn(4, 5)
>>> torch.max(a, b)
tensor([[ 0.9931,  0.3162,  1.7202,  0.2977,  2.4843],
        [ 2.1384,  0.6760,  0.6328, -1.2002,  0.2046],
        [-0.1132,  0.6711,  2.5104,  1.2836, -1.5387],
        [ 0.4333, -0.1055,  2.3136,  1.2815,  0.3704]])