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
I use torch.jit.trace() and save my own CNN_model. The return of my network is a tuple contains 3 tensors.
When I use C++ as front, I use auto output = module.forward(inputs); to get my result. But how can I get the 3 tensors individually? and do some C++ operations?
Thank you!
The text was updated successfully, but these errors were encountered:
script::module's forward method returns an IValue type that has many methods to convert to other types, see here. You might want to try toTensorList(), toTensorVector() or toTuple().
I use
torch.jit.trace()
and save my own CNN_model. The return of my network is a tuple contains 3 tensors.When I use C++ as front, I use
auto output = module.forward(inputs);
to get my result. But how can I get the 3 tensors individually? and do some C++ operations?Thank you!
The text was updated successfully, but these errors were encountered: