| 24 | |
| 25 | template <typename Dtype> |
| 26 | void RNNLayer<Dtype>::RecurrentInputShapes(vector<BlobShape>* shapes) const { |
| 27 | const int num_output = this->layer_param_.recurrent_param().num_output(); |
| 28 | shapes->resize(1); |
| 29 | (*shapes)[0].Clear(); |
| 30 | (*shapes)[0].add_dim(1); // a single timestep |
| 31 | (*shapes)[0].add_dim(this->N_); |
| 32 | (*shapes)[0].add_dim(num_output); |
| 33 | } |
| 34 | |
| 35 | template <typename Dtype> |
| 36 | void RNNLayer<Dtype>::OutputBlobNames(vector<string>* names) const { |