[mlpack-git] [mlpack/mlpack] Inception layer (#757)
Tham
notifications at github.com
Wed Aug 10 03:00:00 EDT 2016
> +void Pad(const arma::Mat<eT>& input, size_t wPad, size_t hPad, arma::Mat<eT>& output)
> +{
> + if (output.n_rows != input.n_rows + wPad * 2 ||
> + output.n_cols != input.n_cols + hPad * 2)
> + output = arma::zeros(input.n_rows + wPad * 2, input.n_cols + hPad * 2);
> + output.submat(wPad, hPad,
> + wPad + input.n_rows - 1,
> + hPad + input.n_cols - 1) = input;
> +}
> +
> +template<typename eT>
> +void Pad(const arma::Cube<eT>& input, size_t wPad, size_t hPad, arma::Cube<eT>& output)
> +{
> + output = arma::zeros(input.n_rows + wPad * 2, input.n_cols + hPad * 2, input.n_slices);
> + for (size_t i = 0; i < input.n_slices; ++i)
> + Pad<double>(input.slice(i), wPad, hPad, output.slice(i));
Why Pad<double> but not Pad<eT>?
---
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/757/files/06d923321f246f2c6ead9ad56e99309fe24a6f5c#r74194353
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160810/89c418a0/attachment.html>
More information about the mlpack-git
mailing list