From 88cf6d9e612cdf893a706d2d9607503b5a5f8094 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Tue, 18 Apr 2017 20:55:59 +0100 Subject: [PATCH] sensible default for unsqueeze --- src/dims/utils.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/dims/utils.jl b/src/dims/utils.jl index 680c85a1..7cc00476 100644 --- a/src/dims/utils.jl +++ b/src/dims/utils.jl @@ -1,4 +1,5 @@ -unsqueeze(xs, dim) = reshape(xs, (size(xs)[1:dim-1]..., 1, size(xs)[dim:end]...)) +unsqueeze(xs, dim = 1) = reshape(xs, (size(xs)[1:dim-1]..., 1, size(xs)[dim:end]...)) +Base.squeeze(xs) = squeeze(xs, 1) stack(xs, dim = 1) = cat(dim, unsqueeze.(xs, dim)...) unstack(xs, dim = 1) = [slicedim(xs, dim, i) for i = 1:size(xs, dim)]