<p>We can also use <code>arma::Mat</code> to store images as <a href="https://github.com/rcurtin" class="user-mention">@rcurtin</a> suggested. I guess the only layer we have to modify is the convolution layer. I'm working on another convolution method that should be faster as the already implemented methods for small kernels. The method uses <code>arma::Mat</code> as input. So, I guess once I finished that, which might take some time, I'll can modify the convolution layer to just use a matrix as input instead of a 3rd-order tensor.</p>

<p>I think we should remove the cube support from the train split method, and use a matrix to store images.</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-196332309">view it on GitHub</a>.<img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFA82zStc-Bj8uJKQqZsR90UK_Tqoks5ptXBVgaJpZM4HhWF3.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-196332309"></link>
  <meta itemprop="name" content="View Pull Request"></meta>
</div>
<meta itemprop="description" content="View this Pull Request on GitHub"></meta>
</div>