[Tensor Subclasses] Trace transfom to interpret __torch_dispatch__ and get the correct output type. based on 1393

[Tensor Subclasses] Trace transfom to interpret __torch_dispatch__ and get the correct output type. based on 1393 #4217