16 lines
464 B
Python
16 lines
464 B
Python
|
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): ...
|
||
|
...
|