Guillaume Lagrange

If the issue really is with the output of your gather node that is a scalar, we need to expand the [`GatherNode`](https://github.com/tracel-ai/burn/blob/main/crates/burn-import/src/burn/node/gather.rs) implementation to support scalar outputs. The codegen in...

Well actually the gather node already handles scalars but only for the index input. You can also check the [`UnsqueezeNode`](https://github.com/tracel-ai/burn/blob/main/crates/burn-import/src/burn/node/unsqueeze.rs).

In Burn, a tensor with a single value is still represented by a 1D tensor with one element. For ONNX, the spec is different so we have to deal with...