#pragma once /* * Copyright (C) 2024 Brett Terpstra * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #ifndef COSC_4P80_ASSIGNMENT_2_LAYER_H #define COSC_4P80_ASSIGNMENT_2_LAYER_H #include #include namespace assign2 { class layer_t { public: layer_t(const blt::i32 in, const blt::i32 out): in_size(in), out_size(out) {} template void init(WeightsFunc weightFunc = empty_init{}, BiasFunc biasFunc = empty_init{}) { weights.resize(in_size, out_size); bias.resize(out_size); weightFunc(weights); biasFunc(bias); } template vector_t call(const vector_t& in, ActFunction func = ActFunction{}) { vector_t out; out.resize(out_size, Eigen::NoChange_t{}); out.noalias() = weights.transpose() * in; out.colwise() += bias; return func(std::move(out)); } private: const blt::i32 in_size, out_size; matrix_t weights{}; matrix_t dweights{}; // derivative of weights vector_t bias{}; vector_t dbias{}; // derivative of bias }; } #endif //COSC_4P80_ASSIGNMENT_2_LAYER_H