Skip to content

Commit b76b0b4

Browse files
authored
【Hackathon 6th No.9】Add cartesian_prod API to Paddle (#6751)
1 parent 1b4715e commit b76b0b4

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

docs/api/paddle/Overview_cn.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,6 +399,7 @@ tensor 元素操作相关(如:转置,reshape 等)
399399

400400
" :ref:`paddle.broadcast_to <cn_api_paddle_broadcast_to>` ", "根据 shape 指定的形状广播 x ,广播后, x 的形状和 shape 指定的形状一致"
401401
" :ref:`paddle.broadcast_tensors <cn_api_paddle_broadcast_tensors>` ", "对一组输入 Tensor 进行广播操作, 输入应符合广播规范"
402+
" :ref:`paddle.cartesian_prod <cn_api_paddle_cartesian_prod>` ", "对输入 Tensor 序列进行笛卡尔积操作"
402403
" :ref:`paddle.cast <cn_api_paddle_cast>` ", "将输入的 x 的数据类型转换为 dtype 并输出"
403404
" :ref:`paddle.chunk <cn_api_paddle_chunk>` ", "将输入 Tensor 分割成多个子 Tensor"
404405
" :ref:`paddle.concat <cn_api_paddle_concat>` ", "对输入沿 axis 轴进行联结,返回一个新的 Tensor"
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
.. _cn_api_paddle_cartesian_prod:
2+
3+
cartesian_prod
4+
-------------------------------
5+
6+
.. py:function:: paddle.cartesian_prod(x, name=None)
7+
8+
9+
对指定的 tensor 序列进行笛卡尔积操作。该行为类似于 python 标准库中的 itertools.product 方法。
10+
相当于将所有的输入 tensors 转换为列表后,对其使用 itertools.product 方法,最终将返回的列表转换为 tensor。
11+
12+
13+
参数
14+
:::::::::
15+
16+
- **x** (list[Tensor]|tuple[Tensor]) – 任意数量的 1-D Tensor 序列,支持的数据类型:bfloat16、float16、float32、float64、int32、int64、complex64、complex128。
17+
18+
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
19+
20+
返回
21+
:::::::::
22+
23+
笛卡尔积运算后的 Tensor,数据类型与输入 Tensor 相同。
24+
25+
代码示例
26+
::::::::::::
27+
28+
COPY-FROM: paddle.cartesian_prod

0 commit comments

Comments
 (0)