MaskIndices

Return indices of a mask of an (n, n) array.

Abstract Signature:

MaskIndices(n: int, mask_func: Callable, k: int = 0)

JAX (Core)

API: jax.numpy.mask_indices
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.mask_indices
Strategy: Direct Mapping
Official Docs ↗