Merge pull request #2 from longriyao/patch-1

camel007 · web-flow · commit 781a7f2ac5ed · 2017-08-09T09:19:02.000+08:00
give the output shape of shuffle
diff --git 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>

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,7 @@ void ShuffleChannelLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,`
`11`	`11`	`group_ = this->layer_param_.shuffle_channel_param().group();`
`12`	`12`	`CHECK_GT(group_, 0) << "group must be greater than 0";`
`13`	`13`	`temp_blob_.ReshapeLike(*bottom[0]);`
	`14`	`+ top[0]->ReshapeLike(*bottom[0]);`
`14`	`15`	`}`
`15`	`16`
`16`	`17`	`template <typename Dtype>`