Onnx Squeeze to accept a list of axes
mtobin-tdab opened this issue · comments
Feature description
https://onnx.ai/onnx/operators/onnx__Squeeze.html
The Squeeze operation to accept a list of axes to squeeze.
Feature motivation
This is an option in ONNX which currently fails when converting:
DEBUG onnx_ir::from_onnx: checking node squeeze1 for constants
DEBUG onnx_ir::from_onnx: checking input Argument { name: "const_axes__420", ty: Tensor(TensorType { elem_type: Int64, dim: 1, shape: Some([2]) }), value: Some(Int64s([2, 3])), passed: false } for const
ERROR burn_import::logger: PANIC => panicked at crates/onnx-ir/src/dim_inference.rs:463:9:
Squeeze must specify only 1 axis, found 2