View source on GitHub
|
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.
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.
|