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