tf.keras.ops.diagflat

Create a two-dimensional array with the flattened input on the k-th diagonal.

View aliases

Main aliases

tf.keras.ops.numpy.diagflat

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

Args
x	Input tensor to be flattened and placed on the diagonal.
k	The diagonal to place the flattened input. Defaults to 0. Use k > 0 for diagonals above the main diagonal, and k < 0 for diagonals below the main diagonal.

Returns
A 2-D tensor with the flattened input on the specified diagonal.