[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