torch.Tensor.as_subclass

Tensor.as_subclass(cls) → Tensor

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