<p>I guess a simple example would make things more clear. Imagine you have a dataset with a bunch of colour images (e.g. CIFAR-10), so more than one channel per image or dataset, something like that:</p>

<p align="center">
<a href="https://camo.githubusercontent.com/88911cfba7b9419c8497f4dc61cf0b18af36ab68/68747470733a2f2f757267732e6f72672f6d61726375732f6d656469612f77655f6e6565645f746f5f676f5f6465657065722e706e67" target="_blank"><img src="https://camo.githubusercontent.com/88911cfba7b9419c8497f4dc61cf0b18af36ab68/68747470733a2f2f757267732e6f72672f6d61726375732f6d656469612f77655f6e6565645f746f5f676f5f6465657065722e706e67" alt="we need to go deeper color" data-canonical-src="https://urgs.org/marcus/media/we_need_to_go_deeper.png" style="max-width:100%;"></a>
</p>

<p>So, I guess I would store that image in a cube with 3 slices one slice per channel. But now I would like to store 10 images in the same data structure that would result in a cube with 10 * 3 slices.</p>

<p>As far as I can see is that your code works with data sets that contains one channel per image, something like that would be totally fine:</p>

<p align="center">
<a href="https://camo.githubusercontent.com/c33cd0ceef812f7dea5f5d3fcbbf3c93b726806f/68747470733a2f2f757267732e6f72672f6d61726375732f6d656469612f77655f6e6565645f746f5f676f5f6465657065725f677261792e706e67" target="_blank"><img src="https://camo.githubusercontent.com/c33cd0ceef812f7dea5f5d3fcbbf3c93b726806f/68747470733a2f2f757267732e6f72672f6d61726375732f6d656469612f77655f6e6565645f746f5f676f5f6465657065725f677261792e706e67" alt="we need to go deeper gray" data-canonical-src="https://urgs.org/marcus/media/we_need_to_go_deeper_gray.png" style="max-width:100%;"></a>
</p>

<p>As I said I'm not sure we should handle such datasets with the code. I guess, we have to specify the number of channels/slices per dataset somehow to make it work.</p>

<p>I hope this makes some sense.</p>

<p style="font-size:small;-webkit-text-size-adjust:none;color:#666;">&mdash;<br>Reply to this email directly or <a href="https://github.com/mlpack/mlpack/pull/523#issuecomment-188945938">view it on GitHub</a>.<img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFKhf0VewWZ6dM43hVX70RkXbl0TEks5pn09CgaJpZM4HhWF3.gif" width="1" /></p>
<div itemscope itemtype="http://schema.org/EmailMessage">
<div itemprop="action" itemscope itemtype="http://schema.org/ViewAction">
  <link itemprop="url" href="https://github.com/mlpack/mlpack/pull/523#issuecomment-188945938"></link>
  <meta itemprop="name" content="View Pull Request"></meta>
</div>
<meta itemprop="description" content="View this Pull Request on GitHub"></meta>
</div>