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