mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Compute mean and variance
This commit is contained in:
@@ -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;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user