torch.Tensor.as_subclass
Tensor.as_subclass(cls) -> Tensor
Makes a "cls" instance with the same data pointer as "self".
   Changes in the output mirror changes in "self", and the output
   stays attached to the autograd graph. "cls" must be a subclass of
   "Tensor".