torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

Преобразует экземпляр cls с тем же указателем на данные, что и self. Изменения в выходных данных отражают изменения в self, и выходные данные остаются привязанными к графу autograd. cls должен быть подклассом Tensor.