- paddle.fluid.layers.tensor. eye ( num_rows, num_columns=None, batch_shape=None, dtype='float32', name=None ) [source]
This function constructs a or a batch of 2-D tensor with ones on the diagonal and zeros elsewhere.
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.
batch_shape (list, optional) – If provided, the returned tensor will have a leading batch size of this shape, the data type of
batch_shapeis int. Default is None.
dtype (np.dtype|str, optional) – The data type of the returned tensor. It should be int32, int64, float16, float32, float64, default 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.
An identity Tensor or LoDTensor of shape batch_shape + [num_rows, num_columns].
- Return type
import paddle.fluid as fluid data = fluid.layers.eye(3, dtype='int32') # [[1, 0, 0] # [0, 1, 0] # [0, 0, 1]] data = fluid.layers.eye(2, 3, dtype='int32') # [[1, 0, 0] # [0, 1, 0]] data = fluid.layers.eye(2, batch_shape=) # Construct a batch of 3 identity tensors, each 2 x 2. # data[i, :, :] is a 2 x 2 identity tensor, i = 0, 1, 2.