is there any ways for me to change the name of input variables of traced module? I have looked into the documentation of torch.jit.trace and it could not be changed beforehand as well, e.g. with dictionary.
basically I’m trying to mimic the behavior of input_names argument in torch.onnx.export