This commit is contained in:
Frank Seide 2016-01-23 21:07:36 -08:00
Родитель 2f640310b6
Коммит 26911b7471
2 изменённых файлов: 2 добавлений и 1 удалений

Просмотреть файл

@ -1724,6 +1724,7 @@ protected:
using Base::GetSampleMatrixNumCols; \
using Base::GetSampleMatrixNumRows; \
using Base::GetTensorShape; \
using Base::GetTensorSliceFor; \
using Base::Gradient; \
using Base::GradientAsMatrix; \
using Base::GradientFor; \

Просмотреть файл

@ -467,7 +467,7 @@ public:
// check/fuse dimensions
if (isFinalValidationPass && dim != dims[k] && dim != 1 && dims[k] != 1)
InvalidArgument("%ls %ls operation: Conflicting dimension %d between %ls %ls operation (%d) and other(s) (%d)",
NodeName().c_str(), OperationName().c_str(), (int)k, Input(i)->NodeName().c_str(), Input(i)->OperationName(), (int)dim, (int)dims[k]);
NodeName().c_str(), OperationName().c_str(), (int)k, Input(i)->NodeName().c_str(), Input(i)->OperationName().c_str(), (int)dim, (int)dims[k]);
if (dims[k] == 1) // broadcast
dims[k] = dim;
}