mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
5b5766665d
See #145101 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/145102 Approved by: https://github.com/bobrenjc93 ghstack dependencies: #145105
20 lines
608 B
Python
20 lines
608 B
Python
from typing import AnyStr, overload
|
|
|
|
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, ...]: ...
|