From d6f2b4630ed807b1fe32803912867d0182900948 Mon Sep 17 00:00:00 2001 From: channings Date: Tue, 29 Oct 2019 16:31:16 +0800 Subject: [PATCH] Update onnx_op_mapper.py --- x2paddle/op_mapper/onnx_op_mapper.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index 4bdfc66..3b8ff04 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -121,7 +121,13 @@ class ONNXOpMapper(OpMapper): for data_node in data_nodes: value_info = value_infos[data_node] - ipt = np.random.random(value_info['shape']).astype( + shape = value_info['shape'] + for i, dim_shape in enumerate(shape): + if dim_shape==0 and i==0: + shape[i]=1 + if dim_shape==0 and i!=0: + assert 'shape of input is not assigned' + ipt = np.random.random(shape).astype( value_info['dtype']) np.save(os.path.join(self.tmp_data_dir, data_node), ipt) -- GitLab