cupy.identity

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

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()