We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 568e059 commit 77f6d41Copy full SHA for 77f6d41
src/caffe/layers/tile_nd_layer.cpp
@@ -14,10 +14,10 @@ void TileNDLayer<Dtype>::LayerSetUp(
14
tile_nd_param.multiples().end(),
15
std::back_inserter(multiples_));
16
17
- CHECK_GT(multiples_.size(), 0) << "Number of tiles must be positive!";
+ CHECK_EQ(multiples_.size(), bottom[0]->num_axes()) << "Length must be the same as the number of dimensions in input!";
18
for(int i=0;i<multiples_.size();i++)
19
{
20
- CHECK_GT(multiples_[i], 0) << "Value of tiles must be positive!";
+ CHECK_GT(multiples_[i], 0) << "Value of multiples must be positive!";
21
}
22
23
0 commit comments