replaced trunc Int with div

This commit is contained in:
Manjunath Bhat 2019-07-16 17:52:55 +05:30 committed by GitHub
parent a128a7718d
commit b779d43aca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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 = trunc(Int, prod(size(x))/channels)
m = div(prod(size(x)), channels)
γ = reshape(BN.γ, affine_shape...)
β = reshape(BN.β, affine_shape...)
if !istraining()