Skip to content

Commit a7bcff1

Browse files
authored
[DLPack] Add new parameters for from_dlpack and move from_dlpack/to_dlpack to top level (#7436)
1 parent b385b69 commit a7bcff1

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

docs/api/paddle/utils/dlpack/from_dlpack_cn.rst renamed to docs/api/paddle/from_dlpack_cn.rst

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
.. _cn_api_paddle_utils_dlpack_from_dlpack:
1+
.. _cn_api_paddle_from_dlpack:
22

33
from_dlpack
44
-------------------------------
55

6-
.. py:function:: paddle.utils.dlpack.from_dlpack(dlpack)
6+
.. py:function:: paddle.from_dlpack(dlpack, *, device=None, copy=None)
77
88
将 DLPack 格式的 Tensor 解码为 Paddle Tensor,DLPACK 是一种通用稳定的内存数据结构,可用于不同深度学习框架之间的 Tensor 共享。
99

@@ -13,6 +13,8 @@ from_dlpack
1313
参数
1414
:::::::::
1515
- **dlpack** (SupportDLPack | PyCapsule) - 一个实现了 ``__dlpack__`` 与 ``__dlpack_device__`` 方法的对象,或者是一个带有 dltensor 的 ``PyCapsule`` 对象。
16+
- **device** (PlaceLike, 可选) - 返回的 Tensor 所在的设备。如果不指定,返回的 Tensor 将与输入的 `dlpack` 位于同一设备上。
17+
- **copy** (bool, 可选) - 是否复制输入的内容。如果为 True,则输出的 Tensor 总是被复制。如果为 False,则输出的 Tensor 永远不会被复制,如果需要复制则会引发 ``BufferError`` 异常。如果为 None,则在可能的情况下重用现有的内存缓冲区,否则进行复制。默认值为 None。
1618

1719
返回
1820
:::::::::

docs/api/paddle/utils/dlpack/to_dlpack_cn.rst renamed to docs/api/paddle/to_dlpack_cn.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
.. _cn_api_paddle_utils_dlpack_to_dlpack:
1+
.. _cn_api_paddle_to_dlpack:
22

33
to_dlpack
44
-------------------------------
55

6-
.. py:function:: paddle.utils.dlpack.to_dlpack(x)
6+
.. py:function:: paddle.to_dlpack(x)
77
88
将 Tensor 对象转化为一个带有 dltensor 的 ``PyCapsule`` 对象,该对象是一种称为 DLPack 的通用稳定内存数据结构,可用于不同深度学习框架之间的 Tensor 共享。
99

0 commit comments

Comments
 (0)