[vulkan] support add for dim < 4 (#41222)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41222
Test Plan: Imported from OSS
Reviewed By: AshkanAliabadi
Differential Revision: D22754937
Pulled By: IvanKobzarev
fbshipit-source-id: f8c5e55c965c0a805e75c63b21f410fb0c323515