NeuralNet.h

Go to the documentation of this file.

#pragma once
 
#include "Tensor.h"
#include "NeuralError.h"
 
#include <tl/expected.hpp>
#include <string>
#include <vector>
#include <memory>
 
namespace NN
{
    class NeuralNet {
    public:
        virtual ~NeuralNet() = default;
 
        virtual tl::expected<std::vector<Tensor>, NeuralError> infer(const Tensor& input) = 0;
 
        tl::expected<std::vector<Tensor>, NeuralError> operator()(const Tensor& input) {
            return infer(input);
        }
 
        virtual size_t inputCount() const = 0;
        virtual size_t outputCount() const = 0;
        virtual Shape inputShape(size_t idx = 0) const = 0;
 
        virtual Shape outputShape(size_t idx = 0) const = 0;
 
        virtual std::string inputName(size_t idx = 0) const = 0;
 
        virtual std::string outputName(size_t idx = 0) const = 0;
 
        virtual int gpuId() const = 0;
    };
 
    using NeuralNetPtr = std::shared_ptr<NeuralNet>;
}