1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from typing import AnyStr, overload, Tuple
from torch import Tensor
class UndefinedGrad:
def __init__(self) -> None: ...
def __call__(self, *inputs: Tensor) -> list[Tensor]: ...
class DelayedError:
def __init__(self, msg: AnyStr, num_inputs: int) -> None: ...
# __call__ should really be a higher-kinded type:
# def __call__(self, arg: Tensor) -> Tensor: ...
# def __call__(self, *args: Tensor * num_inputs) -> Tuple[Tensor * num_inputs]: ...
@overload
def __call__(self, i0: Tensor) -> Tensor: ...
@overload
def __call__(self, *args: Tensor) -> Tuple[Tensor, ...]: ...
|
