linear.hpp

#pragma once
 
#include "../tensor/tensorptr.hpp"
#include "./model.hpp"
 
namespace dl {
    class Linear final : public Model<TensorPtr(TensorPtr)> {
    private:
        TensorPtr weights;
        TensorPtr bias;
 
    public:
        Linear(size_t inFeatures, size_t outFeatures, const Device& device, bool bias = true) noexcept;
        Linear(size_t inFeatures, size_t outFeatures, bool bias = true) noexcept;
 
    public:
        virtual TensorPtr forward(TensorPtr input) noexcept override;
    };
} // namespace dl