import torch from .. import Tensor from typing import Tuple, Optional import builtins class Parameter(Tensor): def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ... ... class UninitializedParameter(Tensor): def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ... def materialize(self, shape: Tuple[int, ...], device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None): ... ...