| 33 | } |
| 34 | |
| 35 | const vector<Tensor> Split::Forward(int flag, const vector<Tensor>& inputs) { |
| 36 | vector<Tensor> outputs; |
| 37 | CHECK_EQ(inputs.size(), 1u) << "Split layer only have one input tensor."; |
| 38 | for (size_t i = 0; i < output_size_; i++) |
| 39 | outputs.push_back(inputs.at(0)); |
| 40 | return outputs; |
| 41 | } |
| 42 | |
| 43 | const std::pair<vector<Tensor>, vector<Tensor>> Split::Backward( |
| 44 | int flag, const vector<Tensor>& grads) { |