torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

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