Skip to content

Implement Scatter ONNX Operation #4175

@antimora

Description

@antimora

Feature description

Implement the ONNX Scatter operation in the burn-import crate, mapping ONNX's Scatter semantics into Burn. The goal is to ensure that models using the Scatter op can be imported and run smoothly in Burn.

Feature motivation

Scatter is available in Burn but not in burn-import. This is a common operation required for efficient tensor manipulation and indexing in modern ONNX models.

(Optional) Suggest a Solution

  • Replicate the approach taken by other indexing-related ops (e.g., Gather) currently implemented in burn-import.
  • Add robust testing for various shapes and indices.

For more information:

Related context: #1714

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions