From 520a871973b899b047c414ea05103a62d681a007 Mon Sep 17 00:00:00 2001 From: dorhar Date: Thu, 26 May 2022 18:03:57 +0300 Subject: [PATCH] add default for keepdims in reduce mean --- onnx2kerastl/operation_layers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnx2kerastl/operation_layers.py b/onnx2kerastl/operation_layers.py index 79541f71..d4beb01a 100644 --- a/onnx2kerastl/operation_layers.py +++ b/onnx2kerastl/operation_layers.py @@ -139,7 +139,8 @@ def convert_reduce_mean(node, params, layers, lambda_func, node_name, keras_name input_0 = ensure_tf_type(layers[node.input[0]], name="%s_const" % keras_name) - keepdims = params['keepdims'] == 1 + param_keepdims = params.get('keepdims',1) + keepdims = param_keepdims == 1 axes = params['axes'] reduce_mean_layer = OnnxReduceMean(axes=axes, keepdims=keepdims)