22 lines
395 B
C++
22 lines
395 B
C++
#pragma once
|
|
|
|
#include "core/omp_config.h"
|
|
|
|
#include "utils/vector.h"
|
|
#include "utils/matrix.h"
|
|
|
|
namespace neural_networks{
|
|
|
|
|
|
template <typename T>
|
|
struct Layer {
|
|
utils::Matrix<T> outputs;
|
|
utils::Matrix<T> dinputs;
|
|
|
|
virtual void forward(const utils::Matrix<T>& inputs) = 0;
|
|
virtual void backward(const utils::Matrix<T>& dvalues) = 0;
|
|
|
|
virtual ~Layer() = default;
|
|
};
|
|
}
|