One hot

[akshitgaur2005]

Pull Request Template

Checklist

• Confirmed that run-checks all script has been executed.
• Made sure the book is up to date with changes in this PR.

Related Issues/PRs

#1544
#1714

Changes

Implement OneHot node for burn_import

Testing

Not yet tested.

This is a draft PR as I cannot figure out how to parse depth field of OneHot, as this is my first time working on such a project.

Any guidance will be helpful.

[laggui]

Btw in case you missed it we have a tensor.one_hot_fill(...) method that was added in #2613 and closely follows the ONNX spec 🙂

[akshitgaur2005]

Oh, I missed it. I can't find its documentation though?

[laggui]

It was added after the last release, so it's not on the hosted documentation yet. But it's here (and in the book).

The up-to-date book on main can be browsed locally if you launch:

cargo xtask books burn open