jax.numpy.tril_indices_from

Warning

This page was created from a pull request (#9655).

jax.numpy.tril_indices_from¶

jax.numpy.tril_indices_from(arr, k=0)[source]¶

Return the indices for the lower-triangle of arr.

LAX-backend implementation of tril_indices_from().

Original docstring below.

See tril_indices for full details.

Parameters

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

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

previous

jax.numpy.tril_indices

next

jax.numpy.trim_zeros