torch.Tensor.as_subclass

Tensor.as_subclass(cls) → 张量

创建一个与 self 具有相同数据指针的 cls 实例。输出中的更改会反映 self 中的更改，并且输出会保持在自动求导图中。cls 必须是 Tensor 的子类。