<p><a href="https://github.com/abhinavchanda" class="user-mention">@abhinavchanda</a>, The code that you have written is only for a linear layer. This needs to be generalized.</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/552#issuecomment-193189629">view it on GitHub</a>.<img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFBrJ8RlmP6-MQ5NpK1UbNygotvhBks5pq_kggaJpZM4HpuO1.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/552#issuecomment-193189629"></link>
  <meta itemprop="name" content="View Pull Request"></meta>
</div>
<meta itemprop="description" content="View this Pull Request on GitHub"></meta>
</div>