tf.keras.ops.tril

Return lower triangle of a tensor.

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

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

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

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