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): ... | |
... |