Method Forward

src/model/layer/softmax.cc:32–38

}

const Tensor Softmax::Forward(int flag, const Tensor& input) {
  CHECK_LE(input.nDim(), 2u);
  Tensor output =  SoftMax(input);
  if (flag & kTrain)
    buf_.push(output);
  return output;
}

const std::pair<Tensor, vector<Tensor>> Softmax::Backward(int flag,
                                                          const Tensor& grad) {