Skip to content

Instantly share code, notes, and snippets.

@Icemole
Created June 29, 2023 09:33
Show Gist options
  • Save Icemole/3eedac24a108f63cb441defb7897d039 to your computer and use it in GitHub Desktop.
Save Icemole/3eedac24a108f63cb441defb7897d039 to your computer and use it in GitHub Desktop.
Exported graph: graph(%d1 : Int(requires_grad=0, device=cpu),
%d2 : Int(requires_grad=0, device=cpu),
%d3 : Int(requires_grad=0, device=cpu)):
%onnx::Unsqueeze_3 : Long(1, strides=[1], device=cpu) = onnx::Constant[value={0}]()
%/Unsqueeze_output_0 : Int(1, strides=[1], device=cpu) = onnx::Unsqueeze[onnx_name="/Unsqueeze"](%d1, %onnx::Unsqueeze_3), scope: __main__.DummyModel::
%onnx::Unsqueeze_5 : Long(1, strides=[1], device=cpu) = onnx::Constant[value={0}]()
%/Unsqueeze_1_output_0 : Int(1, strides=[1], device=cpu) = onnx::Unsqueeze[onnx_name="/Unsqueeze_1"](%d2, %onnx::Unsqueeze_5), scope: __main__.DummyModel::
%onnx::Unsqueeze_7 : Long(1, strides=[1], device=cpu) = onnx::Constant[value={0}]()
%/Unsqueeze_2_output_0 : Int(1, strides=[1], device=cpu) = onnx::Unsqueeze[onnx_name="/Unsqueeze_2"](%d3, %onnx::Unsqueeze_7), scope: __main__.DummyModel::
%/Concat_output_0 : Int(3, strides=[1], device=cpu) = onnx::Concat[axis=0, onnx_name="/Concat"](%/Unsqueeze_output_0, %/Unsqueeze_1_output_0, %/Unsqueeze_2_output_0), scope: __main__.DummyModel:: # /home/nbeneitez/Documentos/work/repos/returnn_pytorch/testtorchonnx.py:25:0
%casted_data : Float(*, *, *, strides=[4, 2, 1], requires_grad=0, device=cpu) = onnx::ConstantOfShape[value={2}, onnx_name="/ConstantOfShape"](%/Concat_output_0), scope: __main__.DummyModel:: # /home/nbeneitez/Documentos/work/repos/returnn_pytorch/testtorchonnx.py:25:0
return (%casted_data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment