Hi, the output_o had been converted back to an at::Tensor, and this is now placed inside pnet.forward().
And error is thrown at this line.
output_o needs to be converted to a torch::jit::IValue an then put inside pnet.forward().
The same goes to rnet.forward() as well.