Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
Support dictionary outputs in TorchScript tracer #27743
@thuyen you can already pass named tuple as output of the tracer, it's just that the result of the traced function will be just plain tuple.
#29751 this will enable tracer to take named tuple as output and preserve the names. But there's on-going discussions/issues on this PR and will not go into the 1.4 release candidate that we are having soon. This will go into the next release after the 1.4 rc.