未验证 提交 7be8cf9c 编写于 作者: W wangchaochaohu 提交者: GitHub

refine doc for chunk (#2559)

上级 26b5f0b9
......@@ -9,7 +9,7 @@ chunk
**参数**:
- **x** (Tensor) - 输入变量,数据类型为bool, float16, float32,float64,int32,int64的多维Tensor。
- **chunks** (int) - 如果 ``chunks`` 是一个整数,则表示Tensor平均划分为相同大小子Tensor的数量
- **chunks** (int) - ``chunks`` 是一个整数,表示将输入Tensor划分成多少个相同大小的子Tensor
- **axis** (int|Tensor,可选) - 整数或者形状为[1]的Tensor,数据类型为int32或int64。表示需要分割的维度。如果 ``axis < 0`` ,则划分的维度为 ``rank(x) + axis`` 。默认值为0。
- **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
......@@ -39,4 +39,4 @@ chunk
out0, out1, out2 = paddle.chunk(x, chunks=3, axis=-2)
# out0.shape [3, 3, 5]
# out1.shape [3, 3, 5]
# out3.shape [3, 3, 5]
# out2.shape [3, 3, 5]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册