cunumeric.tril_indices#

cunumeric.tril_indices(n: int, k: int = 0, m: int | None = None) tuple[ndarray, ...]#

Return the indices for the lower-triangle of an (n, m) array.

Parameters:
  • n (int) – The row dimension of the arrays for which the returned indices will be valid.

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

  • m (int, optional) – The column dimension of the arrays for which the returned indices will be valid. By default m is taken equal to n.

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

Notes

Availability:

Multiple GPUs, Multiple CPUs