ruby-dnn 0.5.1 → 0.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '09358674bd9463c57f3329fe29fc3597e76c7f691bde08097069ca3c82b4c6e9'
-  data.tar.gz: cd26c50e67a70f339d26df1a131a88c46dc77658d32994d92e908472ccea9140
+  metadata.gz: a2197d74209bbc77661471244060d6d2081093db2905bb19c1d4c5c76d1a7ae9
+  data.tar.gz: 6b3d91a99743bacd4852dac9cf129834c0fe2f59b0692c4c6f97c3e067f33b58
 SHA512:
-  metadata.gz: bd189bd96810220d873a3e2ce14c0124147d2e3dd9291914984ecaa5d14b7b7e50f222372958c22739af83590d26defc38071e6c4a90570b987f46c77ad62e07
-  data.tar.gz: 5571c28f7ee17c4464b5a901f0e20aa0526894fa51364f8e0f0c6fc398ce9c0b6a3ba03385877afa924e59a7e82c0a511ce1f528f955162c93756196cc40c858
+  metadata.gz: 510e5b822ff38629fa64c1cdc67c640040d5183e1185b3e817f13969d8361d1ac28b8d889b8f89a0628f88568e4ec8c7233ca317c92cb67816afee96bad50634
+  data.tar.gz: b2bb5f3b6680ad50040bdcda794bbabea88c703ff386776f3d9fdc24555251535a5042f4ecfa1129e21b0c975288dbbdc3ad47384cb4c1c9e660237b20c2fa8d

data/lib/dnn/core/layers.rb

@@ -190,11 +190,12 @@ module DNN
 
       def forward(x)
         @x_shape = x.shape
-        x.reshape(*@shape)
+        x.reshape(x.shape[0], *@shape)
       end
 
       def backward(dout)
-        dout.reshape(@x_shape)
+        p dout.shape, @x_shape
+        dout.reshape(*@x_shape)
       end
 
       def to_hash

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.5.1"
+  VERSION = "0.5.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.5.2
 platform: ruby
 authors:
 - unagiootoro