Modules/DataAnalysis/public/FullyConnectedNN.hpp
2023-10-21 23:20:20 +03:00

20 lines
290 B
C++

#pragma once
#include "Buffer.hpp"
#include "DataAnalysisCommon.hpp"
namespace tp {
class FullyConnectedNN {
public:
struct Layer {
halnf mBias;
Buffer<halnf> mWeights;
};
public:
FullyConnectedNN() = default;
private:
Buffer<Layer> mLayers;
};
};