diag

paddle.fluid.layers.diag(diagonal)[source]

This OP creates a square matrix which has diagonal values specified by input diagonal.

Parameters

diagonal (Variable|numpy.ndarray) – The input tensor should be 1D tensor, the input shape is \([ N]\) , specifying diagonal values by this input tensor. The input data type should be float32, float64, int32, int64.

Returns

The tensor variable storing the square matrix, the diagonal values specified by input diagonal. the output shape is \([N, N]\) with two dims.

Return type

Variable, the output data type is the same as input data type.

Examples

# [[3, 0, 0]
#  [0, 4, 0]
#  [0, 0, 5]

import paddle.fluid as fluid
import numpy as np
diagonal = np.arange(3, 6, dtype='int32')
data = fluid.layers.diag(diagonal)
# diagonal.shape=(3,) data.shape=(3, 3)