Compute mean and variance

This commit is contained in:
2024-04-29 20:55:11 +02:00
parent 0ab623fa23
commit 5c8d3f7e25
6 changed files with 143 additions and 59 deletions

View File

@@ -74,8 +74,12 @@ class BatchNorm : public WeightedLayer {
float* d_output;
float* d_mean;
float* d_mean_sub;
float* d_sqrt_var;
float* d_length;
float* d_epsilon;
float* d_weights;
float* d_biases;