Construct an identity matrix, or a batch of matrices.