torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

创建一个与self具有相同数据指针的cls实例。cls的变化会反映在self上，且输出仍然连接到自动求导图。cls必须是Tensor的一个子类。