Test that multi_tensor optimizer state buffers match with single_tensor state buffers (#81894)
Add testing for state of multitensor optimizers suggested in #78807 (previously only the equality of model parameters after a few optimizer steps was being tested)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81894
Approved by: https://github.com/albanD