Fix bug in caffe2 transpose on GPU (#22233)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/22233
Fix bug in caffe2 transpose on GPU
Reviewed By: hl475
Differential Revision: D15994973
fbshipit-source-id: 542dc8757b51a6322fffa55826c1d4e32927398d