Add support for non square matrices

This commit is contained in:
2024-05-20 15:20:43 +02:00
parent 6f8b5f4081
commit 74098b24e3
21 changed files with 314 additions and 299 deletions

View File

@@ -81,8 +81,8 @@ class Dense : public WeightedLayer {
int getInputSize();
private:
unsigned int inputSize;
unsigned int outputSize;
int inputSize;
int outputSize;
float* d_output;
@@ -95,8 +95,8 @@ class Dense : public WeightedLayer {
Layers::Activation* activation;
// Precompute kernel launch parameters
unsigned int forwardGridSize;
unsigned int biasGridSize;
int forwardGridSize;
int biasGridSize;
/**
* @brief Initialize the weights to zeros