mlx.core.isposinf

目录

mlx.core.isposinf#

isposinf(a: array, stream: None | Stream | Device = None) array#

返回一个布尔数组,指示哪些元素是正无穷大。

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

  • stream (Union[None, Stream, Device]) – 可选的流或设备。

返回值:

表示哪些元素是正无穷大的布尔数组。

返回类型:

array