tf.keras.ops.triu

Return upper triangle of a tensor.

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

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

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.

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