lib/dnn/core/layers.rb in ruby-dnn-0.13.2 vs lib/dnn/core/layers.rb in ruby-dnn-0.13.3

@@ -166,11 +166,11 @@
         super()
         @weight_initializer = weight_initializer
         @bias_initializer = bias_initializer
         @weight_regularizer = weight_regularizer
         @bias_regularizer = bias_regularizer
-        @weight = Param.new(nil, 0)
-        @bias = use_bias ? Param.new(nil, 0) : nil
+        @weight = Param.new(nil, Xumo::SFloat[0])
+        @bias = use_bias ? Param.new(nil, Xumo::SFloat[0]) : nil
       end
 
       def regularizers
         regularizers = []
         regularizers << @weight_regularizer if @weight_regularizer