用户
 找回密码
 立即注册
发表于 2021-4-16 14:39:40
21830
问:想跟您请教下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

使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册