default keep dim is true

AsylumCorp · Jun 19, 2019 · 03a8c41 · 03a8c41
1 parent 1fe0bf1
commit 03a8c41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Source/CNTKv2LibraryDll/proto/onnx/CNTKToONNX.cpp b/Source/CNTKv2LibraryDll/proto/onnx/CNTKToONNX.cpp
@@ -7077,7 +7077,7 @@ void CNTKToONNXHelper::SetReduceElementsAttributes(const FunctionPtr src, Node *
         reductionAxes.push_back((Axis)(src->Attributes()[L"axis"].Value<Axis>()));
 
     //
-    int64_t keepReducedDimensions = 0;
+    int64_t keepReducedDimensions = 1;
     if (src->Attributes().Contains(L"reductionKeepDimensions"))
         keepReducedDimensions = (int64_t)((bool)src->Attributes()[L"reductionKeepDimensions"].Value<bool>() ? 1 : 0);