提交 53e76a2e 编写于 作者: S shippingwang

Fix comments display, test=develop

上级 83f2e2c9
...@@ -9338,6 +9338,7 @@ def get_tensor_from_selected_rows(x, name=None): ...@@ -9338,6 +9338,7 @@ def get_tensor_from_selected_rows(x, name=None):
def shuffle_channel(x, group, name=None): def shuffle_channel(x, group, name=None):
""" """
**Shuffle Channel Operator** **Shuffle Channel Operator**
This operator shuffles the channels of input x. This operator shuffles the channels of input x.
It divide the input channels in each group into :attr:`group` subgroups, It divide the input channels in each group into :attr:`group` subgroups,
and obtain a new order by selecting element from every subgroup one by one. and obtain a new order by selecting element from every subgroup one by one.
...@@ -9346,6 +9347,7 @@ def shuffle_channel(x, group, name=None): ...@@ -9346,6 +9347,7 @@ def shuffle_channel(x, group, name=None):
https://arxiv.org/pdf/1707.01083.pdf https://arxiv.org/pdf/1707.01083.pdf
.. code-block:: text .. code-block:: text
Given a 4-D tensor input with the shape (N, C, H, W): Given a 4-D tensor input with the shape (N, C, H, W):
input.shape = (1, 4, 2, 2) input.shape = (1, 4, 2, 2)
input.data =[[[[0.1, 0.2], input.data =[[[[0.1, 0.2],
...@@ -9387,7 +9389,8 @@ def shuffle_channel(x, group, name=None): ...@@ -9387,7 +9389,8 @@ def shuffle_channel(x, group, name=None):
Examples: Examples:
.. code-block:: python .. code-block:: python
input = fluid.layers.data(name='input', shape=[1,4,2,2], dtype='float32')
input = fluid.layers.data(name='input', shape=[4,2,2], dtype='float32')
out = fluid.layers.shuffle_channel(x=input, group=2) out = fluid.layers.shuffle_channel(x=input, group=2)
""" """
helper = LayerHelper("shuffle_channel", **locals()) helper = LayerHelper("shuffle_channel", **locals())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册