diff --git a/doc/fluid/api_cn/tensor_cn/chunk_cn.rst b/doc/fluid/api_cn/tensor_cn/chunk_cn.rst index d4c29bc421c7b4d287a99d51eac22e82c1bc9165..b7e9f0381a88065a387a407c6fbb2f940ba080d2 100644 --- a/doc/fluid/api_cn/tensor_cn/chunk_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/chunk_cn.rst @@ -28,7 +28,7 @@ chunk x_np = np.random.random([3, 9, 5]).astype("int32") x = paddle.to_tensor(x_np) - out0, out1, out22 = paddle.chunk(x, chunks=3, axis=1) + out0, out1, out2 = paddle.chunk(x, chunks=3, axis=1) # out0.shape [3, 3, 5] # out1.shape [3, 3, 5] # out2.shape [3, 3, 5] @@ -39,3 +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] diff --git a/doc/fluid/api_cn/tensor_cn/numel_cn.rst b/doc/fluid/api_cn/tensor_cn/numel_cn.rst index 360dad0e10f2359bcc971d9a6fbfb60bb2e0a1e9..c22bee48074fc0c538f137f1ab6c5b56d8205864 100644 --- a/doc/fluid/api_cn/tensor_cn/numel_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/numel_cn.rst @@ -9,7 +9,7 @@ numel 该OP返回一个长度为1并且元素值为输入 ``x`` 元素个数的Tensor。 参数: - - **x** (Tensor) - 输入Tensor,数据类型为int32,int64, float16, float32, float64, int32, int64 。 + - **x** (Tensor) - 输入Tensor,数据类型为float16, float32, float64, int32, int64 。 返回: 返回长度为1并且元素值为 ``x`` 元素个数的Tensor。