torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

创建一个 cls 实例，该实例与 self 共享相同的数据指针。输出中的变化会反映 self 中的变化，并且输出仍附加到 autograd 图中。cls 必须是 Tensor 的子类。