问：想跟您请教下torch.randperm这个算子转onnx不支持应该怎...

GPUS-Cat

问：想跟您请教下torch.randperm这个算子转onnx不支持应该怎么实现？
答复：可以先用 ONNX_FALLTHROUGH转成onnx，然后再给TensorRT写个plugin，转出来后也可以用ONNX-graphsurgeon把这个node展开成TRT支持的node（如果可以等效替换掉的话）。
注意：好像没法用TRT layer表示randperm：https://pytorch.org/docs/stable/generated/torch.randperm.html