为什么使用argmax plugin？

你好，之前跑tensorrt官网上提供的https://github.com/NVIDIA/TensorRT/tree/release/8.5/quickstart/SemanticSegmentation 例子中，也使用了argmax，pth模型转onnx转tensorrt，可以直接可以使用；想问下，程序中有使用argmax plugin，为什么会使用argmax plugin，或者说使用argmax plugin的目的是什么？希望能帮解答一下，多谢