identity<T extends SizedNativeType> function

VARP identity<T extends SizedNativeType>(
  1. int n
)

identity(n, dtype=float32) Return the identity var. The identity var 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 float.

Returns

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

Examples

np.identity(3) var([1., 0., 0., 0., 1., 0.,

 [0.,  0.,  1.]])

Implementation

VARP identity<T extends ffi.SizedNativeType>(int n) {
  return eye<T>(n);
}