diff --git a/src/utils.jl b/src/utils.jl index afe926d9..bba3e416 100644 --- a/src/utils.jl +++ b/src/utils.jl @@ -4,8 +4,6 @@ initn(dims...) = randn(dims...)/100 glorot_uniform(dims...) = (rand(dims...) - 0.5)*sqrt(24.0/(sum(dims))) glorot_normal(dims...) = (randn(dims...)*sqrt(2.0/sum(dims))) -flatten(xs) = reshape(xs, size(xs, 1), :) - unsqueeze(xs, dim) = reshape(xs, (size(xs)[1:dim-1]..., 1, size(xs)[dim:end]...)) stack(xs, dim) = cat(dim, unsqueeze.(xs, dim)...)