DenseLayer.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef DENSELAYER_HPP_INCLUDE
#define DENSELAYER_HPP_INCLUDE
 
#include "TensorOneD.hpp"
 
namespace geopm
{
    class TensorTwoD;
 
    class DenseLayer
    {
        public:
            static std::unique_ptr<DenseLayer> make_unique(const TensorTwoD &weights,
                                                           const TensorOneD &biases);
            virtual ~DenseLayer() = default;
            virtual TensorOneD forward(const TensorOneD &input) const = 0;
            virtual size_t get_input_dim() const = 0;
            virtual size_t get_output_dim() const = 0;
 
            TensorOneD operator()(const TensorOneD &input) const;
    };
}
 
#endif /* DENSELAYER_HPP_INCLUDE */