search

webonnx / wonnx

A WebGPU-accelerated ONNX inference run-time written 100% in Rust, ready for native and the web
Other
1.54k stars 54 forks source link

Add argmax operator. #175

Open JulienSiems opened 11 months ago

JulienSiems commented 11 months ago

This PR adds the argmax operator to wonnx.