set axis default to 0 (#854)

This commit is contained in:
Friedrich Carl Eichenroth 2023-05-29 22:15:28 +02:00 committed by GitHub
parent 3b158f7a5f
commit 6f2b3755ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -147,6 +147,7 @@ def get_run_onnx(onnx_model):
if n.op_type == "Pow": ret = inp[0] ** inp[1]
elif n.op_type == "Split":
if 'split' not in opt: opt['split'] = [int(x) for x in safe_numpy(inp[1])] # split can be a tensor
if 'axis' not in opt: opt['axis'] = 0
i = 0
arg = [(0,x) for x in inp[0].shape]
for o,s in zip(n.output, opt['split']):