[mlpack-git] [mlpack/mlpack] fixes: convolve with correct weight slice, pad input in forward pass (#737)

Marcus Edel notifications at github.com
Mon Aug 8 18:51:34 EDT 2016


> +  conv3.InputParameter() = input;
> +  conv3.Forward(conv3.InputParameter(), conv3.OutputParameter());
> +  Test(conv3.OutputParameter(), output);
> +  
> +  arma::cube error = arma::ones(5, 5, 1);
> +  conv3.Backward(conv3.InputParameter(), error, conv3.Delta());
> +  Test(conv3.Delta(), arma::ones(5, 5, 4));
> +
> +  arma::cube delta = arma::zeros(5, 5, 1);
> +  delta(2, 2, 0) = 1;
> +  conv3.Gradient(conv3.InputParameter(), delta, conv3.Gradient());
> +  arma::cube gradOut(3, 3, 4);
> +  arma::mat grad_w;
> +  grad_w << 7 << 8 << 9 << arma::endr
> +        << 12 << 13 << 14 << arma::endr
> +        << 17 << 18 << 19;

The rest looks good, thanks!

---
You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
https://github.com/mlpack/mlpack/pull/737/files/e8ef7f47ff7d398bd8a7a6982860294a5a468f68#r73970749
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160808/cda66313/attachment.html>


More information about the mlpack-git mailing list