DenseLayerImp.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef DENSELAYERIMP_HPP_INCLUDE
#define DENSELAYERIMP_HPP_INCLUDE
 
#include "DenseLayer.hpp"
 
#include "TensorOneD.hpp"
#include "TensorTwoD.hpp"
 
namespace geopm
{
    class DenseLayerImp : public DenseLayer
    {
        public:
            DenseLayerImp(const TensorTwoD &weights, const TensorOneD &biases);
            TensorOneD forward(const TensorOneD &input) const override;
            size_t get_input_dim() const override;
            size_t get_output_dim() const override;
 
        private:
            TensorTwoD m_weights;
            TensorOneD m_biases;
    };
}
 
#endif /* DENSELAYERIMP_HPP_INCLUDE */