[Vulkan] Use 2D tensor views when possible (#57198)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57198
Test Plan: Imported from OSS
Reviewed By: IvanKobzarev
Differential Revision: D28293561
Pulled By: SS-JIA
fbshipit-source-id: 77099cfe54870a19e926a067de9852787eb55d0b