From 2b0f5207488fdbff045ae03310af9e064fed66e0 Mon Sep 17 00:00:00 2001
From: abh2k <41710346+abh2k@users.noreply.github.com>
Date: Thu, 22 Jul 2021 10:36:51 +0530
Subject: [PATCH] Update nearest_interpolation_impl.hpp

---
 src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp b/src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp
index 159e138ed0..fe7d269123 100644
--- a/src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp
+++ b/src/mlpack/methods/ann/layer/nearest_interpolation_impl.hpp
@@ -101,6 +101,7 @@ void NearestInterpolation<InputDataType, OutputDataType>::Backward(
   if (output.is_empty())
   {
     output.set_size(inRowSize * inColSize * depth, batchSize);
+    output.zeros();
   }
   else
   {