torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

使用与 self 相同的数据指针创建一个 cls 实例。输出中的更改会反映在 self 中，并且输出保持连接到自动梯度图。 cls 必须是 Tensor 的子类。