未验证 提交 3da0e06f 编写于 作者: J Jason 提交者: GitHub

Merge pull request #792 from wjj19950828/Fixed_ConvTranspose

Fixed ONNX ConvTranspose
......@@ -2354,7 +2354,8 @@ class OpSet9():
output_size = node.get_attr('output_shape', [])
pads = node.get_attr('pads', [0] * (convnd * 2))
paddings, var_x = self._pad_if_asymmetric(node, pads, val_x)
paddings = np.array(pads).reshape((2, -1)).transpose().astype("int32")
paddings = paddings.flatten().tolist()
if len(output_size) != 0:
paddings = [0] * 4
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册