From 56ce10a4fd0a7fb8f21dbff3b937040e65f89d98 Mon Sep 17 00:00:00 2001 From: Elian Dib <elian.dib@inria.fr> Date: Fri, 21 Sep 2018 14:57:58 +0200 Subject: [PATCH] Improved squeeze functionality --- split.m | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/split.m b/split.m index 091d7af..4896c75 100644 --- a/split.m +++ b/split.m @@ -37,10 +37,12 @@ dimDists = arrayfun(@(nsplit,sz) repelem(sz/nsplit,nsplit),... slices = mat2cell(data,dimDists{:}); if squeezeGrid - slices = squeeze(slices); + gridSize = dimsSize(dimsToSplit); + slices = reshape(slices,gridSize); end if squeezeSlices - slices = cellfun(@squeeze,slices,'UniformOutput',false); + SliceSize = dimsSize(~dimsToSplit); SliceSize(end+1:2)=1; + slices = cellfun(@(s) reshape(s,SliceSize),slices,'UniformOutput',false); end end \ No newline at end of file -- GitLab