def eye(n, cls=None):
 if (cls is None):
  from sympy.matrices import Matrix as cls
 return cls.eye(n)

def eye ( n , cls = None ) : if ( cls is None ) : from sympy . matrices import Matrix as cls return cls . eye ( n )
return a 2-d array with ones on the diagonal and zeros elsewhere .