eye(n, m=None, optimizable=False, dtype=None)

Source code in funfact/lang/_special.py

def eye(n, m=None, optimizable=False, dtype=None):
    return TsrEx(
        P.tensor(
            AbstractTensor(
                n, m or n, symbol=('I', None),
                initializer=ini.Eye(dtype=dtype),
                optimizable=optimizable
            )
        )
    )