Creating a Pluging layer in TensorRT

I want to create an ArgMax layer plugin. I read the trt samples, but I don’t know how to do that! Do you have any other tutorial or example about creating a plugin layer in trt?



Please reference the following examples for extending TensorRT functionalities by implementing custom layers using the IPluginV2 class for the C++ and Python API.