triton.language.maximum

triton.language.maximum(x, y, propagate_nan: ~triton.language.core.constexpr = <PROPAGATE_NAN.NONE: 0>)

计算 xy 的逐元素最大值。

参数
  • x (Block) – 第一个输入张量

  • y (Block) – 第二个输入张量

  • propagate_nan (tl.PropagateNan) – 是否传播 NaN 值。

另请参阅

tl.PropagateNan