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/tutorial

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.