jax.scipy.linalg.tril

jax.scipy.linalg.tril(m, k=0)[source]

Make a copy of a matrix with elements above the kth diagonal zeroed.

LAX-backend implementation of tril().

Original docstring below.

Parameters

  • m (array_like) – Matrix whose elements to return

  • k (int, optional) – Diagonal above which to zero elements. k == 0 is the main diagonal, k < 0 subdiagonal and k > 0 superdiagonal.

Returns

tril – Return is the same shape and type as m.

Return type

ndarray