I’d say that for  clone  and  data  you probably want to preserve your own Tensor class, instead of it being de-promoted back to Tensor class.

I copied the behaviour of torch.nn.Parameter class, it actually de-promotes to Tensor (that seems to be expected).