_functions.pyi 608 B

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