[pytree] Allow register_pytree_node to take in 5 inputs (#108256)
Summary: This is currently breaking internal old torch.packages from when _register_pytree_node took 5 inputs.
Test Plan: CI
Differential Revision: D48834742
Pull Request resolved: https://github.com/pytorch/pytorch/pull/108256
Approved by: https://github.com/zhxchen17, https://github.com/zou3519
