NVIDIA已经把GatherND和ScatterND发布出来了,python调用的例子在这里:https://github.com/NVIDIA/trt-samples-for-hackathon-cn/tree/master/python