o iN@sDdZddlZddlmZddlZeejdejdejfddZdS) N)wrapstensorreturncKstj|fii|ddi}|}dkr:tjd|tjtjBd}tj|tjd}| |d|j | | | dd|S)N requires_gradFr)prot)dtype)torch empty_likeuntyped_storagenbytesmmap PROT_READ PROT_WRITE frombufferuint8set_shapestriderequires_grad_get)rkwargsemptyr bufferrK/home/ubuntu/.local/lib/python3.10/site-packages/spaces/zero/torch/utils.pyempty_like_raw_alloc sr)__doc__r functoolsrr r Tensorrrrrrs