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 f048fc7 commit cc7f7abCopy full SHA for cc7f7ab
src/caffe/layers/space_to_batch_nd_layer.cpp
@@ -62,6 +62,8 @@ void SpaceToBatchNDLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
62
const vector<Blob<Dtype>*>& top) {
63
const Dtype* bottom_data = bottom[0]->cpu_data();
64
Dtype* top_data = top[0]->mutable_cpu_data();
65
+ const int count = top[0]->count();
66
+ caffe_set(count, Dtype(0), top_data);
67
vector<int> bottom_shape = bottom[0]->shape();
68
vector<int> top_shape = bottom_shape;
69
0 commit comments