diff --git a/doc/fluid/api_cn/tensor_cn/chunk_cn.rst b/doc/fluid/api_cn/tensor_cn/chunk_cn.rst index b7e9f0381a88065a387a407c6fbb2f940ba080d2..da78f9e2951ff2e4fce59a3ab55b087ce9d87f94 100644 --- a/doc/fluid/api_cn/tensor_cn/chunk_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/chunk_cn.rst @@ -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]