Files
pytorch/torch/utils/dlpack.py
2017-10-21 20:19:51 +02:00

5 lines
108 B
Python

import torch
from torch._C import _from_dlpack as from_dlpack
from torch._C import _to_dlpack as to_dlpack