!44914 Fix doc error of nn.ChannelShuffle

Merge pull request !44914 from pkuliuliu/code_docs_channel_shuffle
This commit is contained in:
i-robot 2022-11-01 03:33:55 +00:00 committed by Gitee
commit 2bab0a4c55
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions

View File

@ -1,7 +1,7 @@
mindspore.nn.ChannelShuffle
============================
.. py:class:: mindspore.nn.ChannelShuffle()
.. py:class:: mindspore.nn.ChannelShuffle(groups)
将shape的为 :math`(*, C, H, W)` 的Tensor的通道划分成 :math`g` 组,并将其以 :math`(*, C \frac g, g, H, W)` 的shape重新排列, 同时保持Tensor原有的shape。

View File

@ -36,7 +36,7 @@ class ChannelShuffle(Cell):
Tensor, with the same type and shape as the `x`.
Raises:
TypeError: If groups is not an int.
TypeError: If `groups` is not an int.
ValueError: If `groups` is less than 1.
ValueError: If dims of `x` is less than 3.
ValueError: If number of channels can not be divisible by groups.