This commit is contained in:
Nikos Karampatziakis 2017-02-02 11:30:50 -08:00
Родитель cb4ef4243d
Коммит 81c97d24bd
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -559,7 +559,7 @@ namespace CNTK
}
case PrimitiveOpType::ReduceElements:
{
assert(inputs.size() == 1);
assert(m_inputs.size() == 1);
auto reductionAxis = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameAxis].Value<Axis>(), m_inputs[0].Shape());
if (reductionAxis == Axis::AllStaticAxes() || reductionAxis == Axis::AllAxes())
outputShape = {};