jax.numpy.triu

jax.numpy.triu(m, k=0)[source]

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.