Convert c10::IValue to tensor or tuple failed

[yf225 (PyTorch Developer, Meta)]

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