how can I use torch.nn.functional.conv1d in tensorRT？

[yulinhuyang]

Description

I use torch.nn.functional.conv1d(img, query) as a module in torch, but it can not convert to a onnx/tensorRT module successfully,so how can I implement it in tensorRT。

Environment

TensorRT Version: 6.1 ONNX-TensorRT Version / Branch: GPU Type: Nvidia Driver Version: CUDA Version: 10 CUDNN Version: 7.6 Operating System + Version: Python Version (if applicable): 3.6 PyTorch Version (if applicable): 1.4

[kevinch-nv]

Can you provide a model that you have converted?