Warning

This page was created from a pull request.

jax.numpy.tril_indices_from

jax.numpy.tril_indices_from(arr, k=0)

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).0