nn 2.3.0 → 2.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dae759f927b41bad2f29c7d4739cf198cf0080e7054d2e8276d518db94bb33f3
-  data.tar.gz: 99594955a3e1d8801a2415413387cc9ca86b068a25e3f079ce15d19226318952
+  metadata.gz: c0c39df6ef1009aa84fa49e626213fe36f4ec72dc87050651ebe37060ccb2d88
+  data.tar.gz: 85e5ddf31d9ee27cb4b86cbd4955dc9410852e0a4c9e115df2485cf267ab05b1
 SHA512:
-  metadata.gz: 2612677d78320574f714002006b8be85f5abe5d0b0eb107dbb83d8be67eebc99f077c62771325f4d06e372ba54e4a60f9f1c7edbb5bea3214a610790502e9b0e
-  data.tar.gz: 36268d75d8b43bf9d3fc7442d4d3b15ca93f45bb585e0fd73f6967586ba576ebbadbeed957ef59c731ce76daf40159441c6684c26238e3e6e53a1ca659e7fc7f
+  metadata.gz: 9e8155689f6996cb4beaa2cb41ce0879c8157ea3a1c4825a0c1367f3042d8f7ff5a25c18d89be8dc1a62a0ff1ed6949031148884bfb3b2e3cb0fd5ef44dfd0a4
+  data.tar.gz: 81fc0f1e73b9f5a9d91b227d33b5a09426f4329299211f3fd6ffa0eeede953ab5d744b0632104781a64b1555be9eb41b41fa5b77356e03a62848b90040a1f285

data/document.txt

@@ -153,4 +153,4 @@ http://d.hatena.ne.jp/n_shuyo/20090913/mnist
 2018/6/10 バージョン2.0.1公開
 2018/6/17 バージョン2.1.0公開
 2018/6/17 バージョン2.2.0公開
-2018/6/24 バージョン2.3.0公開
+2018/6/24 バージョン2.4.0公開

data/lib/nn.rb

@@ -2,7 +2,7 @@ require "numo/narray"
 require "json"
 
 class NN
-  VERSION = "2.3"
+  VERSION = "2.4"
 
   include Numo
 
@@ -424,8 +424,8 @@ class NN::BatchNorm
   end
 
   def backward(dout)
-    @d_beta = dout.sum(0).mean
-    @d_gamma = (@xn * dout).sum(0).mean
+    @d_beta = dout.sum(0)
+    @d_gamma = (@xn * dout).sum(0)
     dxn = @nn.gammas[@index] * dout
     dxc = dxn / @std
     dstd = -((dxn * @xc) / (@std ** 2)).sum(0)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nn
 version: !ruby/object:Gem::Version
-  version: 2.3.0
+  version: 2.4.0
 platform: ruby
 authors:
 - unagiootoro