fix an possible issue of reshape for TileND

haifenghan · haifenghan · commit 568e05922f17 · 2019-06-24T15:29:14.000+08:00
diff --git a/src/caffe/layers/tile_nd_layer.cpp b/src/caffe/layers/tile_nd_layer.cpp
@@ -27,7 +27,8 @@ void TileNDLayer<Dtype>::Reshape(
     const vector<Blob<Dtype>*>& bottom, const vector<Blob<Dtype>*>& top) {
   vector<int> top_shape = bottom[0]->shape();
   for(int i=multiples_.size()-1;i>=0;i--)
-    top_shape[i] = bottom[0]->shape()[i] * multiples_[i];
+    if (multiples_[i] > 1)
+      top_shape[i] = bottom[0]->shape()[i] * multiples_[i];
   top[0]->Reshape(top_shape);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,8 @@ void TileNDLayer<Dtype>::Reshape(`
`27`	`27`	`const vector<Blob<Dtype>>& bottom, const vector<Blob<Dtype>>& top) {`
`28`	`28`	`vector<int> top_shape = bottom[0]->shape();`
`29`	`29`	`for(int i=multiples_.size()-1;i>=0;i--)`
`30`		`- top_shape[i] = bottom[0]->shape()[i] * multiples_[i];`
	`30`	`+ if (multiples_[i] > 1)`
	`31`	`+ top_shape[i] = bottom[0]->shape()[i] * multiples_[i];`
`31`	`32`	`top[0]->Reshape(top_shape);`
`32`	`33`	`}`
`33`	`34`