| 13 | explicit ParameterLayer(const LayerParameter& param) |
| 14 | : Layer<Dtype>(param) {} |
| 15 | virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom, |
| 16 | const vector<Blob<Dtype>*>& top) { |
| 17 | if (this->blobs_.size() > 0) { |
| 18 | LOG(INFO) << "Skipping parameter initialization"; |
| 19 | } else { |
| 20 | this->blobs_.resize(1); |
| 21 | this->blobs_[0].reset(new Blob<Dtype>()); |
| 22 | this->blobs_[0]->Reshape(this->layer_param_.parameter_param().shape()); |
| 23 | } |
| 24 | top[0]->Reshape(this->layer_param_.parameter_param().shape()); |
| 25 | } |
| 26 | virtual void Reshape(const vector<Blob<Dtype>*>& bottom, |
| 27 | const vector<Blob<Dtype>*>& top) { } |
| 28 | virtual inline const char* type() const { return "Parameter"; } |