Реализация функции ReLU на C ++ - PullRequest
Новая
       10

Реализация функции ReLU на C ++

0 голосов
/ 06 июля 2018

Я на самом деле работаю над CNN, я использую сигмоид для функции активации, но я бы хотел использовать ReLU.

Я реализовал код для ReLU с использованием Eigen, но, похоже, он не работает, не могли бы вы мне помочь, пожалуйста?

Вот мой код: 

Matrix ReLu(const Matrix & x){

    Matrix A;
    for( int i = 0; i< x.rows(); ++i )
    for (int j=0; i< x.cols(); j++) {
        if (x(i,j) <= 0){
            A(i,j)=(0.0);
        }
        else A(i,j)=(x(i,j));
    }
    return std::move(A.matrix());
}

Matrix ReLu_deriv (const Matrix& y) {
    Matrix B;
    for( int i = 0; i < y.rows(); ++i )
    for (int j=0; i < y.cols() ; j++)
    {
        {
        if (y(i,j) <= 0.0){
            B(i,j)=(0.0);
        }
        else B(i,j)=(1.0);
    }
    return std::move(B.matrix());
}

и ошибка: 

> /usr/include/eigen3/Eigen/src/Core/DenseCoeffsBase.h :365 : Eigen::DenseCoeffsBase<Derived,
> 1>::Scalar& Eigen::DenseCoeffsBase<Derived,
> 1>::operator()(Eigen::Index, Eigen::Index) [with Derived =
> Eigen::Matrix<double, -1, -1>; Eigen::DenseCoeffsBase<Derived,
> 1>::Scalar = double; Eigen::Index = long int]:  l'assertion « row >= 0
> && row < rows() && col >= 0 && col < cols() » a échoué.

Ответы [ 2 ]

0 голосов
/ 06 июля 2018

Сначала вам нужно правильно инициализировать временную матрицу: 

Matrix ReLu(const Matrix & x){

    Matrix A(x.rows(), x.cols());
    for( int i = 0; i< x.rows(); ++i )
    for (int j=0; j< x.cols(); ++j) {
        if (x(i,j) <= 0){
            A(i,j)=(0.0);
        }
        else A(i,j)=(x(i,j));
    }
    return std::move(A.matrix());
}

Matrix ReLu_deriv (const Matrix& y) {
    Matrix B(y.rows(), y.cols());
    for( int i = 0; i < y.rows(); ++i )
    for (int j=0; j < y.cols() ; ++j)
    {
        if (y(i,j) <= 0.0){
            B(i,j)=(0.0);
        }
        else B(i,j)=(1.0);
    }
    return std::move(B.matrix());
}
0 голосов
/ 06 июля 2018

В обоих циклах у вас есть опечатка: 

for( int i = 0; i< x.rows(); ++i )
    for (int j=0; i< x.cols(); j++) { \\ replace i with j
---------------------------------------------------------
for( int i = 0; i< x.rows(); ++i )
    for (int j=0; i< x.cols(); j++) { \\replace i with j
...