Update src/mlpack/methods/ann/layer/log_softmax.hpp
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
co-authored by
Ryan Curtin
parent
a45dcb0327
commit
4f42aaee63
@@ -70,7 +70,7 @@ class LogSoftMaxType : public Layer<MatType>
|
||||
*/
|
||||
void ForwardImpl(const MatType& input, MatType& output,
|
||||
const typename std::enable_if_t<
|
||||
arma::is_arma_type<MatType>::value>* = 0);
|
||||
arma::is_arma_type<MatType>::value>* = 0);
|
||||
|
||||
#ifdef MLPACK_HAS_COOT
|
||||
void ForwardImpl(const MatType& input, MatType& output,
|
||||
|
||||
Reference in New Issue
Block a user