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

Update nearest_interpolation_impl.hpp

Browse Source
This commit is contained in:
abh2k
2021-07-22 10:36:51 +05:30
committed by GitHub
parent 12efd37432
commit 2b0f520748
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp
+1
View File
@@ -101,6 +101,7 @@ void NearestInterpolation<InputDataType, OutputDataType>::Backward(
if (output.is_empty())
{
output.set_size(inRowSize * inColSize * depth, batchSize);
output.zeros();
}
else
{