mlx.core.isneginf

目录

mlx.core.isneginf#

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

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

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

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

返回:

指示哪些元素是负无穷大的布尔数组。

返回类型:

array