mlx.core.maximum

目录

mlx.core.maximum#

maximum(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) array#

按元素取最大值。

使用 numpy 风格的广播语义,按元素获取两个数组的最大值。输入数组可以是一个或两个,也可以是标量。

参数:
  • a (array) – 输入数组或标量。

  • b (array) – 输入数组或标量。

返回值:

`a` 和 `b` 的最大值。

返回类型:

array