lib/dnn/core/layers.rb in ruby-dnn-0.10.2 vs lib/dnn/core/layers.rb in ruby-dnn-0.10.3

@@ -186,9 +186,12 @@
               weight_regularizer: weight_regularizer, bias_regularizer: bias_regularizer, use_bias: use_bias)
         @num_nodes = num_nodes
       end
 
       def build(input_shape)
+        unless input_shape.length == 1
+          raise DNN_ShapeError.new("Input shape is #{input_shape}. But input shape must be 1 dimensional.")
+        end
         super
         num_prev_nodes = input_shape[0]
         @weight.data = Xumo::SFloat.new(num_prev_nodes, @num_nodes)
         @bias.data = Xumo::SFloat.new(@num_nodes) if @bias
         init_weight_and_bias