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 `PReLU` for backend `Gorgonia` #203

Open modanesh opened 1 year ago

modanesh commented 1 year ago

Why is this operator needed?

This operator is required to the QMagFace with the Gorgonia backend.

Implementation

Link to existing material on the backend

Although the PReLU does not exist in the backend, the closest to it is LeakyReLU: Godoc of the operator LeakyReLU on Gorgonia

Expected problems?

Tests