mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
c47f680086
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13141 This is an example diff to show what lint rules are being applied. Reviewed By: mingzhe09088 Differential Revision: D10858478 fbshipit-source-id: cbeb013f10f755b0095478adf79366e7cf7836ff
29 lines
725 B
Python
29 lines
725 B
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
import torch
|
|
|
|
from torch._C import _from_dlpack as from_dlpack
|
|
from torch._C import _to_dlpack as to_dlpack
|
|
|
|
torch._C._add_docstr(from_dlpack, r"""from_dlpack(dlpack) -> Tensor
|
|
|
|
Decodes a DLPack to a tensor.
|
|
|
|
Args:
|
|
dlpack: a PyCapsule object with the dltensor
|
|
|
|
The tensor will share the memory with the object represented
|
|
in the dlpack.
|
|
Note that each dlpack can only be consumed once.
|
|
""")
|
|
|
|
torch._C._add_docstr(to_dlpack, r"""to_dlpack(tensor) -> PyCapsule
|
|
|
|
Returns a DLPack representing the tensor.
|
|
|
|
Args:
|
|
tensor: a tensor to be exported
|
|
|
|
The dlpack shares the tensors memory.
|
|
Note that each dlpack can only be consumed once.
|
|
""")
|