mindspore.ops.Eye
- class mindspore.ops.Eye[source]
Creates a tensor with ones on the diagonal and zeros in the rest.
Note
Combines ReverseV2 operator to get an anti-diagonal Tensor, but ReverseV2 only supports Ascend and GPU platforms currently.
- Inputs:
n (int) - The number of rows of returned tensor. Constant value only.
m (int) - The number of columns of returned tensor. Constant value only.
t (mindspore.dtype) - MindSpore’s dtype, The data type of the returned tensor. The data type can be Number.
- Outputs:
Tensor, a tensor with ones on the diagonal and the rest of elements are zero. The shape of output depends on the user’s Inputs n and m. And the data type depends on Inputs t.
- Raises
TypeError – If m or n is not an int.
ValueError – If m or n is less than 1.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> eye = ops.Eye() >>> output = eye(2, 2, mindspore.int32) >>> print(output) [[1 0] [0 1]] >>> print(output.dtype) Int32 >>> output = eye(1, 2, mindspore.float64) >>> print(output) [[1. 0.]] >>> print(output.dtype) Float64