apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Removed copying

Browse Source
This commit is contained in:
abh2k
2021-02-23 13:55:23 +05:30
committed by GitHub
parent 7b70a820f8
commit 5e695f99c8
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/pixel_shuffle_impl.hpp
+6 -9
View File
@@ -61,12 +61,10 @@ void PixelShuffle<InputDataType, OutputDataType>::Forward(
output.zeros(outputHeight * outputWidth * sizeOut, batchSize);
for (size_t n = 0; n < batchSize; n++)
{
arma::mat inputImage = input.col(n);
arma::mat outputImage = output.col(n);
arma::cube inputTemp(const_cast<arma::mat&>(inputImage).memptr(), height,
width, size, false, false);
arma::cube outputTemp(const_cast<arma::mat&>(outputImage).memptr(),
outputHeight, outputWidth, sizeOut, false, false);
arma::cube inputTemp(const_cast<arma::mat&>(input).memptr(), height,
width, size * batchSize, false, false);
arma::cube outputTemp(const_cast<arma::mat&>(output).memptr(),
outputHeight, outputWidth, sizeOut * batchSize, false, false);
for (size_t c = 0; c < sizeOut ; c++)
{
@@ -78,13 +76,12 @@ void PixelShuffle<InputDataType, OutputDataType>::Forward(
size_t width_index = w / upscaleFactor;
size_t channel_index = (upscaleFactor * (h % upscaleFactor)) +
(w % upscaleFactor) + (c * std::pow(upscaleFactor, 2));
outputTemp(w, h, c) = inputTemp(width_index, height_index,
channel_index);
outputTemp(w, h, c + n * sizeOut) = inputTemp(width_index, height_index,
channel_index + n * size);
}
}
}
output.col(n) = outputImage;
}
}