From e5a6da96225cc402fd75803a44a129233ffbf2b6 Mon Sep 17 00:00:00 2001 From: Frank Seide Date: Tue, 6 Oct 2015 08:50:17 -0700 Subject: [PATCH] bug fix: mixed-up dimensions in TransposeNode::Validate() --- .../CNTKComputationNetworkLib/LinearAlgebraNodes.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/MachineLearning/CNTKComputationNetworkLib/LinearAlgebraNodes.h b/MachineLearning/CNTKComputationNetworkLib/LinearAlgebraNodes.h index 44e1323c9..7ed90146e 100644 --- a/MachineLearning/CNTKComputationNetworkLib/LinearAlgebraNodes.h +++ b/MachineLearning/CNTKComputationNetworkLib/LinearAlgebraNodes.h @@ -2488,9 +2488,9 @@ private: { Base::Validate(isFinalValidationPass); - size_t rows0 = Inputs(0)->GetNumRows(); // , cols0 = Inputs(0)->GetNumCols(); + size_t rows0 = Inputs(0)->GetNumRows(), cols0 = Inputs(0)->GetNumCols(); - Resize(Inputs(0)); + Resize(cols0, rows0); mOnes = Matrix::Ones(rows0, rows0, m_deviceId); m_pMBLayout = nullptr; // this node does not hold mini-batch data InferImageDimsFromInputs();