o پi@s8dZddlmZddlmZddZddZd d Zd S) zDECORD Pytorch bridge)absolute_import) _from_dlpackc Cs>d}ztddgdWSty}z|s|t|d}~ww)zqTry import torch at runtime. Returns ------- torch module if found. Raise ImportError otherwise z>torch is required, you can install by pip: `pip install torch`ztorch.utils.dlpackobject)fromlistN) __import__ ImportError)messageer I/home/ubuntu/.local/lib/python3.10/site-packages/decord/bridge/torchdl.pytry_import_torchsr cCst}||S)zFrom decord to torch. The tensor will share the memory with the object represented in the dlpack. Note that each dlpack can only be consumed once.)r from_dlpack to_dlpack) decord_arrdlpackr r r to_torchrcCst}t||S)ztFrom torch to decord. The dlpack shares the tensors memory. Note that each dlpack can only be consumed once.)r rr)tensorrr r r from_torchrrN)__doc__ __future__r_ffi._ctypes.ndarrayrr rrr r r r s