[mlpack] softmax_layer.hpp

Shangtong Zhang zhangshangtong.cpp at qq.com
Fri Jan 30 02:16:15 EST 2015


Hi Marcus,


I have a question about code in line 48, softmax_layer.hpp
void FeedForward(const VecType inputActivation, VecType outputActivation)
 {
  outputActivation = arma::trunc_exp(inputActivation);
  outputActivation /= arma::accu(inputActivation);
 }


I think it should beoutputActivation /= arma::accu(outputActivation);




Best
Regards


Shangtong Zhang,
Third Year Undergraduate,
School of Computer Science,
Fudan University, China.
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mailman.cc.gatech.edu/pipermail/mlpack/attachments/20150130/11ae1aca/attachment-0002.html>


More information about the mlpack mailing list