!44167 modify pooling docs

Merge pull request !44167 from 冯一航/code_docs_modify_3d
This commit is contained in:
i-robot 2022-10-19 06:16:33 +00:00 committed by Gitee
commit 86e1b47ba2
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions

View File

@ -390,7 +390,7 @@ class AvgPool3d(Cell):
Outputs:
Tensor, with shape :math:`(N, C, D_{out}, H_{out}, W_{out})` or
:math:`(C, D_{in}, H_{in}, W_{in})`, with the same data type with `x`.
:math:`(C, D_{in}, H_{in}, W_{in})`, with the same data type as `x`.
Raises:
TypeError: If `kernel_size`, `stride` or `padding` is neither an int nor a tuple.