layer.h

Go to the documentation of this file.

 
#ifndef NNLIB_LAYER_H
#define NNLIB_LAYER_H
 
#include "tensor.h"
#include <string>
 
class Layer {
public:
    DataLocation location;
 
    size_t outSize;
 
    size_t inSize;
 
    std::string activation;
 
    sTensor weights;
 
    sTensor biases;
 
    Layer(size_t inSize, size_t outSize, std::string activation, DataLocation location);
 
    ~Layer();
 
    // You might want to ignore the return value of forward, so don't use [[nodiscard]]
    // NOLINTNEXTLINE(modernize-use-nodiscard)
    sTensor forward(const sTensor& batch) const;
 
    void applyGradients(size_t batchSize, float learningRate = 0.01);
};
 
#endif //NNLIB_LAYER_H