[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: <https://mailman.cc.gatech.edu/pipermail/mlpack/attachments/20150130/11ae1aca/attachment.html>
More information about the mlpack
mailing list