torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

使用与 self 相同的数据指针创建 cls 实例。输出中的更改会反映在 self 中，并且输出保持附加到 autograd 图。cls 必须是 Tensor 的子类。