分类目录:《深入浅出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]])