Fix the missing dynamic axis for CustomProxyOp

This commit is contained in:
Jaliya Ekanayake 2018-05-24 15:15:13 -07:00
Родитель f49c174f2d
Коммит 6bd9d286ea
1 изменённых файлов: 3 добавлений и 2 удалений

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

@ -270,8 +270,9 @@ namespace CNTK
{
InvalidArgument("Output shape must be specified for CustomProxyOp.");
}
outputs.push_back(OutputVariable(outputShape, outputDataType, m_inputs[0].DynamicAxes(), false, Name().empty() ? L"" : Name()));
std::vector<Axis> outputDynamicAxes = GetOutputDynamicAxes(m_op, m_inputs, this, m_attributes);
outputs.push_back(OutputVariable(outputShape, outputDataType, outputDynamicAxes, false, Name().empty() ? L"" : Name()));
}
else
{