symjax.tensor.linalg.tril

symjax.tensor.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

Examples

>>> from scipy.linalg import tril
>>> tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1)
array([[ 0,  0,  0],
       [ 4,  0,  0],
       [ 7,  8,  0],
       [10, 11, 12]])