Warning

This page was created from a pull request.

jax.numpy.triu

jax.numpy.triu(m, k=0)

Upper triangle of an array.

LAX-backend implementation of triu(). Original docstring below.

Return a copy of a matrix with the elements below the k-th diagonal zeroed.

Please refer to the documentation for tril for further details.

tril : lower triangle of an array

>>> np.triu([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1)
array([[ 1,  2,  3],
       [ 4,  5,  6],
       [ 0,  8,  9],
       [ 0,  0, 12]])