Is there something that plays the role of torch.multinomial in tensorRT?
I couldn’t find it on the site below.
A clear and concise description of the bug or issue.
TensorRT Version: 8.6
GPU Type: rtx A6000
Nvidia Driver Version: 528.49
CUDA Version: 12.0
CUDNN Version: X
Operating System + Version:
Python Version (if applicable): 3.9
TensorFlow Version (if applicable):
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):
Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)
- Exact steps/commands to build your repro
- Exact steps/commands to run your repro
- Full traceback of errors encountered