You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,
How to convert nvinfer1::ITensor into at::tensor? Like #146
What you have already tried
I want to do some operations use libtorch on the nvinfer1::ITensor. So, can I convert nvinfer1::ITensor into at::tensor? Or I must write a custom converter with the libtorch function? @xsacha@aaronp24@itsliupeng@lukeyeager@elezar
The text was updated successfully, but these errors were encountered:
You cant really convert ITensors into at::Tensors, ITensors are just edges between layers in the tensorrt and have no concrete value. You can set that ITensor as an output and feed it to some Torch code outside of the TRTorch program
❓ Question
Hi,
How to convert nvinfer1::ITensor into at::tensor? Like #146
What you have already tried
I want to do some operations use libtorch on the nvinfer1::ITensor. So, can I convert nvinfer1::ITensor into at::tensor? Or I must write a custom converter with the libtorch function?
@xsacha @aaronp24 @itsliupeng @lukeyeager @elezar
The text was updated successfully, but these errors were encountered: