From 49f8fdb3ea1b7de9e116a37ba77ffd8f312f6c5c Mon Sep 17 00:00:00 2001 From: Haritha Date: Thu, 28 Jun 2018 12:37:31 +0530 Subject: [PATCH] simplify forward --- src/mlpack/methods/ann/layer/subview.hpp | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/src/mlpack/methods/ann/layer/subview.hpp b/src/mlpack/methods/ann/layer/subview.hpp index 5bc24cbc15..d6a60942c5 100644 --- a/src/mlpack/methods/ann/layer/subview.hpp +++ b/src/mlpack/methods/ann/layer/subview.hpp @@ -88,17 +88,8 @@ class Subview { for (size_t i = 0; i < batchSize; i++) { - if (beginRow == 0 && endRow == (input.n_rows - 1)) - { - // Matrix of contiguous elements. - output.col(i) = arma::mat(&input(batchBegin * input.n_rows), - (input.n_rows * (batchEnd - batchBegin + 1)), 1, false); - } - else - { - output.col(i) = arma::vectorise( - input.submat(beginRow, batchBegin, endRow, batchEnd)); - } + output.col(i) = arma::vectorise( + input.submat(beginRow, batchBegin, endRow, batchEnd)); // Move to next batch. batchBegin += inSize;