Fix tensor printing (#38031)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38031
Test Plan: Imported from OSS
Differential Revision: D21502915
Pulled By: anjali411
fbshipit-source-id: 0cc3017a390da55af47ba81f651a883cd52b10da
