27 lines
475 B
C++
27 lines
475 B
C++
#pragma once
|
|
|
|
#include "Buffer.hpp"
|
|
#include "DataAnalysisCommon.hpp"
|
|
|
|
namespace tp {
|
|
class FullyConnectedNN {
|
|
struct Layer {
|
|
struct Neuron {
|
|
halnf mBias = 0;
|
|
Buffer<halnf> mWeights;
|
|
halnf mActivationValue = 0;
|
|
};
|
|
|
|
Buffer<Neuron> mNeurons;
|
|
};
|
|
|
|
public:
|
|
FullyConnectedNN() = default;
|
|
|
|
void initializeRandom(Buffer<halni> description);
|
|
void evaluate(const Buffer<halnf>& input, Buffer<halnf>& output);
|
|
|
|
private:
|
|
Buffer<Layer> mLayers;
|
|
};
|
|
};
|