simplify forward

This commit is contained in:
Haritha
2018-06-28 12:37:31 +05:30
parent f6ae5589ae
commit 49f8fdb3ea
+2 -11
View File
@@ -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;