Detect signed infinities in mlx arrays

mlx_isposinf() and mlx_isneginf() mirror mlx.core.isposinf() and mlx.core.isneginf(), returning boolean masks of positive or negative infinities.

Usage

mlx_isposinf(x)

mlx_isneginf(x)

Arguments

x

An mlx array.

Value

An mlx boolean array highlighting infinite entries.

See also

mlx.core.isposinf, mlx.core.isneginf

Examples

vals <- as_mlx(c(-Inf, -1, 0, Inf))
mlx_isposinf(vals)
#> mlx array [4]
#>   dtype: bool
#>   device: gpu
#>   values:
#> [1] FALSE FALSE FALSE  TRUE
mlx_isneginf(vals)
#> mlx array [4]
#>   dtype: bool
#>   device: gpu
#>   values:
#> [1]  TRUE FALSE FALSE FALSE