eye¶

paddle. eye ( num_rows, num_columns=None, dtype=None, name=None ) [source]

This function constructs 2-D Tensor with ones on the diagonal and zeros elsewhere.

Parameters

• num_rows (int) – the number of rows in each batch Tensor.

• num_columns (int, optional) – the number of columns in each batch Tensor. If None, default: num_rows.

• dtype (np.dtype|str, optional) – The data type of the returned Tensor. It should be int32, int64, float16, float32, float64. Default: if None, the data type is float32.

• name (str, optional) – The default value is None. Normally there is no need for user to set this property. For more information, please refer to Name

Returns

An identity Tensor or LoDTensor of shape [num_rows, num_columns].

Return type

Tensor

Examples

import paddle

data = paddle.eye(3, dtype='int32')
# [[1 0 0]
#  [0 1 0]
#  [0 0 1]]
data = paddle.eye(2, 3, dtype='int32')
# [[1 0 0]
#  [0 1 0]]