tf.keras.ops.triu

Return upper triangle of a tensor.

View aliases

Main aliases

tf.keras.ops.numpy.triu

tf.keras.ops.triu(
    x, k=0
)

For tensors with ndim exceeding 2, triu will apply to the final two axes.

Args
x	Input tensor.
k	Diagonal below which to zero elements. Defaults to 0. the main diagonal. k < 0 is below it, and k > 0 is above it.

Returns
Upper triangle of x, of same shape and data type as x.