Convert c10::IValue to tensor or tuple failed

Have you tried output.toTensorList() and then get the Tensors by indexing the returned list?

3 Likes