I’m working on some wrappers for Pytorch functions with the goal of extending the Tensor class, and am running into an issue with functions like
torch.mode (among many others) that return function-specific types like
I’d like to replace the
values attribute of the output object with a different tensor before returning it, but the output is immutable and I can’t find the constructors for the classes in
Any idea where these constructors are located (assuming they’re implemented somewhere in Python)? I’d be alright with returning a plain tuple instead, but would love if there is a means of handling this that’s more faithful to the original Pytorch behavior. Thanks!