Torch.fx usage question

when I try to tansform the order of the operations in my module, I find the graph changed but the forward function remains thesame and it is said that ‘UnboundLocalError: local variable ‘getitem_5’ referenced before assignment’
I just chage the args of some node but it failed in performe a result

Could you post a minimal, executable code snippet which would reproduce this issue, please?