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<Tensor, NeuralError> infer(const Tensor& input) = 0;
 
        tl::expected<Tensor, NeuralError> operator()(const Tensor& input) {
            return infer(input);
        }
 
        virtual Shape inputShape() const = 0;
 
        virtual Shape outputShape() const = 0;
 
        virtual int gpuId() const = 0;
    };
 
    using NeuralNetPtr = std::shared_ptr<NeuralNet>;
}