pytorch

1
import builtins
2
from typing import Optional, Tuple
3

4
import torch
5
from torch import Tensor
6

7
class Parameter(Tensor):
8
    def __init__(
9
        self,
10
        data: Tensor = ...,
11
        requires_grad: builtins.bool = ...,
12
    ): ...
13

14
def is_lazy(param: Tensor): ...
15

16
class UninitializedParameter(Tensor):
17
    def __init__(
18
        self,
19
        data: Tensor = ...,
20
        requires_grad: builtins.bool = ...,
21
    ): ...
22
    def materialize(
23
        self,
24
        shape: Tuple[int, ...],
25
        device: Optional[torch.device] = None,
26
        dtype: Optional[torch.dtype] = None,
27
    ): ...
28

29
class UninitializedBuffer(Tensor):
30
    def __init__(
31
        self,
32
        data: Tensor = ...,
33
        requires_grad: builtins.bool = ...,
34
    ): ...
35
    def materialize(
36
        self,
37
        shape: Tuple[int, ...],
38
        device: Optional[torch.device] = None,
39
        dtype: Optional[torch.dtype] = None,
40
    ): ...
41