diff --git a/src/layer/shufflechannel.cpp b/src/layer/shufflechannel.cpp index f478d1858..47fea2426 100644 --- a/src/layer/shufflechannel.cpp +++ b/src/layer/shufflechannel.cpp @@ -41,8 +41,8 @@ int ShuffleChannel::forward(const Mat &bottom_blob, Mat &top_blob) const size_t feature_sz = w * h * sizeof(float); for (int i = 0; i != group; ++i) { for (int j = 0; j != chs_per_group; ++j) { - dst_q = chs_per_group * i + j; - src_q = chs_per_group * j + i; + src_q = chs_per_group * i + j; + dst_q = group * j + i; memcpy(top_blob.channel(dst_q), bottom_blob.channel(src_q), feature_sz); }