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
704 stars 72 forks source link

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

Closed godshen closed 10 months ago

godshen commented 10 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