From 34d905dc342d171c5e8d3c382aa9cc67fdfc57ec Mon Sep 17 00:00:00 2001
From: longriyao <jixiaolong888@gmail.com>
Date: Thu, 13 Jul 2017 18:32:46 +0800
Subject: [PATCH] give the output shape of shuffle

---
 shuffle_channel_layer.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/shuffle_channel_layer.cpp b/shuffle_channel_layer.cpp
index 9351f6c..4b3b09a 100644
--- a/shuffle_channel_layer.cpp
+++ b/shuffle_channel_layer.cpp
@@ -11,6 +11,7 @@ void ShuffleChannelLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
     group_ = this->layer_param_.shuffle_channel_param().group();
     CHECK_GT(group_, 0) << "group must be greater than 0";
     temp_blob_.ReshapeLike(*bottom[0]);
+    top[0]->ReshapeLike(*bottom[0]);
 }
 
 template <typename Dtype>