search

oramasearch / onnx-go

onnx-go gives the ability to import a pre-trained neural network within Go without being linked to a framework or library.
https://blog.owulveryck.info/2019/04/03/from-a-project-to-a-product-the-state-of-onnx-go.html
MIT License
717 stars 71 forks source link

Implement operator `Gather` for backend `YYY` #214

Closed godshen closed 11 months ago

godshen commented 11 months ago

Why is this operator needed?

this operator is needed to run the model bert generated from pytorch.onnx.export with the backend

Implementation

official document https://pytorch.org/docs/stable/generated/torch.gather.html