From 3ef72a9d7b19b48fa01e089a3af889704b93ca72 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 6 Sep 2017 18:59:43 -0400 Subject: [PATCH] utils updates --- src/compiler/loops.jl | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/compiler/loops.jl b/src/compiler/loops.jl index f3a2b67b..62fe0533 100644 --- a/src/compiler/loops.jl +++ b/src/compiler/loops.jl @@ -1,8 +1,4 @@ -unsqueeze(xs, dim = 1) = reshape(xs, (size(xs)[1:dim-1]..., 1, size(xs)[dim:end]...)) -squeeze(xs, dim = 1) = Base.squeeze(xs, dim) - -stack(xs, dim = 1) = cat(dim, unsqueeze.(xs, dim)...) -unstack(xs, dim = 1) = [slicedim(xs, dim, i) for i = 1:size(xs, dim)] +using ..Flux: stack, unstack, squeeze, unsqueeze # Stateful Models