examples/api-examples/initializer_example.rb in ruby-dnn-0.16.2 vs examples/api-examples/initializer_example.rb in ruby-dnn-1.0.0

@@ -34,10 +34,10 @@
     @bn1 = BatchNormalization.new
     @bn2 = BatchNormalization.new
   end
 
   def forward(x)
-    x = InputLayer.(x)
+    x = InputLayer.new(784).(x)
     x = @l1.(x)
     x = @bn1.(x)
     x = ReLU.(x)
     x = @l2.(x)
     x = @bn2.(x)