Name	Size	Mode	Actions
activation.cpython-36.pyc	47635	0644	editdlrm
adaptive.cpython-36.pyc	9725	0644	editdlrm
batchnorm.cpython-36.pyc	31039	0644	editdlrm
channelshuffle.cpython-36.pyc	1901	0644	editdlrm
container.cpython-36.pyc	27970	0644	editdlrm
conv.cpython-36.pyc	57789	0644	editdlrm
distance.cpython-36.pyc	3684	0644	editdlrm
dropout.cpython-36.pyc	10310	0644	editdlrm
flatten.cpython-36.pyc	5813	0644	editdlrm
fold.cpython-36.pyc	12807	0644	editdlrm
instancenorm.cpython-36.pyc	18808	0644	editdlrm
lazy.cpython-36.pyc	11690	0644	editdlrm
linear.cpython-36.pyc	10188	0644	editdlrm
loss.cpython-36.pyc	91465	0644	editdlrm
module.cpython-36.pyc	66530	0644	editdlrm
normalization.cpython-36.pyc	11360	0644	editdlrm
padding.cpython-36.pyc	22217	0644	editdlrm
pixelshuffle.cpython-36.pyc	4398	0644	editdlrm
pooling.cpython-36.pyc	53203	0644	editdlrm
rnn.cpython-36.pyc	44653	0644	editdlrm
sparse.cpython-36.pyc	20956	0644	editdlrm
transformer.cpython-36.pyc	20737	0644	editdlrm
upsampling.cpython-36.pyc	10780	0644	editdlrm
utils.cpython-36.pyc	2528	0644	editdlrm
_functions.cpython-36.pyc	5441	0644	editdlrm
__init__.cpython-36.pyc	5236	0644	editdlrm

3 ûEgã@s8dd
lm
Z

ddlmZ
ddlmZ
Gdd„de
ƒZdS) é
)
ÚModuleé)
Ú functionalé)
ÚTensorcsReZ
dZUd
Zdg
Zeeddœ‡f
dd„Zeedœdd „Z e d œ
dd„Z‡ZS) ÚChannelShufflea/Divide the channels in a tensor of shape :math:`(*, C , H, W)` into g groups and rearrange them as :math:`(*, C \frac g, g, H, W)`, while keeping the original tensor shape. Args: groups (int): number of groups to divide channels in. Examples:: >>> channel_shuffle = nn.ChannelShuffle(2) >>> input = torch.randn(1, 4, 2, 2) >>> print(input) [[[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]], [[13, 14], [15, 16]], ]] >>> output = channel_shuffle(input) >>> print(output) [[[[1, 2], [3, 4]], [[9, 10], [11, 12]], [[5, 6], [7, 8]], [[13, 14], [15, 16]], ]] ÚgroupsN)rÚreturncstt
|ƒjƒ
|
|_dS)
N)ÚsuperrÚ__init__r)Úselfr)
Ú __class__©úK/usr/local/lib64/python3.6/site-packages/torch/nn/modules/channelshuffle.pyr,s

zChannelShuffle.__init__)Úinputr cCstj
|
|jƒS)
N)Ú
FZchannel_shuffler)rrrrrÚforward0s
zChannelShuffle.forward)
r c

Csd
j|j
ƒ
S)Nz groups={})Úformatr)
rrrrÚ extra_repr3s
zChannelShuffle.extra_repr) Ú__name__Ú __module__Ú__qualname__Ú__doc__Z __constants__ÚintrrrrÚstrrÚ __classcell__rr)
r rrs !

rN)ÚmodulerÚrrZtorchrrrrrrÚ
s