torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

创建一个与 cls 具有相同数据指针的 self 实例。输出变化与 self 同步，并且输出保持与自动微分图关联。 cls 必须是 Tensor 的子类。