LocalNeuralNet.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef LOCALNEURALNET_HPP_INCLUDE
#define LOCALNEURALNET_HPP_INCLUDE
 
#include "DenseLayer.hpp"
#include "TensorOneD.hpp"
 
namespace geopm
{
    class TensorTwoD;
 
 
    class LocalNeuralNet
    {
        public:
            virtual ~LocalNeuralNet() = default;
            static std::unique_ptr<LocalNeuralNet> make_unique(std::vector<std::shared_ptr<DenseLayer> > layers);
            virtual TensorOneD forward(const TensorOneD &inp) 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  /* LOCALNEURALNET_HPP_INCLUDE */