From 2816fbb9b24572549fe9ff48909dc825ad7346bf Mon Sep 17 00:00:00 2001
From: thebhatman <manjunathbhat9920@gmail.com>
Date: Fri, 12 Jul 2019 22:19:41 +0530
Subject: [PATCH] Fix for getindex error in BatchNorm

---
 src/layers/normalise.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl
index 59b39ca7..2876cdd7 100644
--- a/src/layers/normalise.jl
+++ b/src/layers/normalise.jl
@@ -136,7 +136,7 @@ function (BN::BatchNorm)(x)
   dims = length(size(x))
   channels = size(x, dims-1)
   affine_shape = ntuple(i->i == ndims(x) - 1 ? size(x, i) : 1, ndims(x))
-  m = prod(size(x)[1:end-2]) * size(x)[end]
+  m = trunc(Int, prod(size(x))/channels)
   γ = reshape(BN.γ, affine_shape...)
   β = reshape(BN.β, affine_shape...)
   if !istraining()