Method Backward

src/model/layer/flatten.cc:49–54

}

const std::pair<Tensor, vector<Tensor> > Flatten::Backward(int flag,
                                                           const Tensor &grad) {
  vector<Tensor> param_grad;
  Tensor input_grad = Reshape(grad, input_shape_);
  return std::make_pair(input_grad, param_grad);
}

} // namespace singa