问:想跟您请教下torch.randperm这个算子转onnx不支持应该怎么实现? 答复:可以先用 ONNX_FALLTHROUGH转成onnx,然后再给TensorRT写个plugin,转出来后也可以用ONNX-graphsurgeon把这个node展开成TRT支持的node(如果可以等效替换掉的话)。 注意:好像没法用TRT layer表示randperm:torch.randperm — PyTorch 1.11.0 documentation |
有人也给了其他思路,比如:用random.shuffle 改写下!
参考:
PyTorch | torch.randperm()使用方法_SophiaCV的博客-CSDN博客_pytorch中randperm