diff --git a/crates/burn-import/onnx-tests/tests/pad/pad.onnx b/crates/burn-import/onnx-tests/tests/pad/pad.onnx index 4c8c265c42c..ddd8b8a5aa3 100644 Binary files a/crates/burn-import/onnx-tests/tests/pad/pad.onnx and b/crates/burn-import/onnx-tests/tests/pad/pad.onnx differ diff --git a/crates/burn-import/src/onnx/op_configuration.rs b/crates/burn-import/src/onnx/op_configuration.rs index 36ab799ce90..d28ce725764 100644 --- a/crates/burn-import/src/onnx/op_configuration.rs +++ b/crates/burn-import/src/onnx/op_configuration.rs @@ -785,8 +785,8 @@ pub fn pad_config(node: &Node) -> PadConfig { let bottom_index = pads.len() - 2; let index_list = [left_index, top_index, right_index, bottom_index]; - for index in 0..pads.len() { - if !index_list.contains(&index) && pads[index] != 0 { + for (index, &item) in pads.iter().enumerate() { + if !index_list.contains(&index) && item != 0 { panic!("Pad: padding will only be applied to the last two dimensions but found non zero padding for other dimensions"); } }