identity

identity(n, dtype=None, ctx=None)

Return the identity array.

The identity array is a square array with ones on the main diagonal.

Parameters:

n : int

Number of rows (and columns) in n x n output.

dtype : data-type, optional

Data-type of the output. Defaults to numpy.float32.

ctx : Context, optional

An optional device context (default is the current default context).

Returns:

out : ndarray

n x n array with its main diagonal set to one, and all other elements 0.

Examples

>>> np.identity(3)
>>> np.identity(3)
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])