!44914 Fix doc error of nn.ChannelShuffle
Merge pull request !44914 from pkuliuliu/code_docs_channel_shuffle
This commit is contained in:
commit
2bab0a4c55
|
@ -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。
|
||||
|
||||
|
|
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue