Skip to content

Commit c576467

Browse files
authored
add_dtensor_from_fn_cn_doc (#6130)
* add_dtensor_from_fn_cn_doc * resolve issues raised by code review * fix rst grammar bugs * Modify according to ZhongKai's review * delete unnecessary links * pre-commit approve
1 parent 414d815 commit c576467

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
.. _cn_api_paddle_distributed_dtensor_from_fn:
2+
3+
dtensor_from_fn
4+
-------------------------------
5+
6+
.. py:class:: paddle.distributed.dtensor_from_fn(fn, dist_attr, *args, **kwargs)
7+
8+
通过一个 paddle API (一般是 Tensor 创建类的 API )结合分布式属性 dist_attr 创建一个带分布式属性的 Tensor。
9+
10+
参数
11+
:::::::::
12+
13+
- **fn** - paddle 公开的可创建 Tensor 的 API。例如: :ref:`cn_api_paddle_empty` 、 :ref:`cn_api_paddle_ones` 、:ref:`cn_api_paddle_zeros` 等 paddle API。
14+
- **dist_attr** (paddle.distributed.DistAttr) - 描述 Tensor 在 ProcessMesh 上的分布或切片方式。
15+
- ***args** - fn 函数的输入参数( Tuple 形式)
16+
- ****kwargs** - fn 函数的输入参数( Dict 形式)
17+
18+
19+
返回
20+
:::::::::
21+
带有分布式信息的 Tensor
22+
23+
24+
25+
**代码示例**
26+
27+
COPY-FROM: paddle.distributed.dtensor_from_fn

0 commit comments

Comments
 (0)