[functorch] Fixes printing issue with TensorWrapper (pytorch/functorch#397)
* Fixes printing issue with TensorWrapper
Description:
- Hacky fix to fix printing tensors inside `grad`
Fixes pytorch/functorch#382
* Applied comments from the review and updated test
* Added printing test from the issue
* Updated tests
* Added dtypes and bdim