torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

建立一個 cls 例項，該例項與 self 共享相同的資料指標。輸出中的變化會反映 self 中的變化，並且輸出仍附加到 autograd 圖中。cls 必須是 Tensor 的子類。