def I_like(A: LinearOperator) -> Identity:
    """ A function that produces an Identity operator with the same
        shape, dtype and device as A """
    Op = Identity(dtype=A.dtype, shape=A.shape)
    Op.to(A.device)
    return Op
 A function that produces an Identity operator with the same
        shape, dtype and device as A 
cola/ops/operators.py