4 ms·

Implementing a custom layer isn't hard[0], that said if it were me I'd rather add it in the runtime, e.g. via a TensorRT plugin. [0] e.g. https://jamesmccaffre

• mathisfun123 3y ago
  That tut only shows you how to remix existing operators. That obviously won't work here (not what's required). You need to do this https://pytorch.org/tutorials/advanced/dispatcher.html https://pytorch.org/tutorials/advanced/dispatcher.html More complex but still not that hard.