[vulkan] fix broadcasting (#81631)
fixed broadcasting issue where it was only selecting one or the other's size
Differential Revision: [D37923488](https://our.internmc.facebook.com/intern/diff/D37923488/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81631
Approved by: https://github.com/SS-JIA