cupy.identity

cupy.identity(n, dtype=<class 'float'>)

Returns a 2-D identity array.

It is equivalent to eye(n, n, dtype).

Parameters:

• n (int) – Number of rows and columns.
• dtype – Data type specifier.

Returns:

A 2-D identity array.

Return type:

cupy.ndarray

See also

numpy.identity()