cunumeric.tril_indices_from

cunumeric.tril_indices_from(arr: ndarray, k: int = 0) → tuple[ndarray, ...]

Return the indices for the lower-triangle of arr.

See cunumeric.tril_indices() for full details.

Parameters:

• arr (array_like) – The indices will be valid for arrays whose dimensions are the same as arr.

• k (int, optional) – Diagonal offset (see cunumeric.tril() for details).

Returns:

inds – The indices for the lower-triangle. The returned tuple contains two arrays, each with the indices along one dimension of the array.

Return type:

tuple of arrays

See also

numpy.tril_indices_from

Notes

Availability:

Multiple GPUs, Multiple CPUs