_functions.pyi 345 B

123456789101112
  1. from torch import Tensor
  2. from typing import AnyStr, List
  3. class UndefinedGrad:
  4. def __init__(self) -> None: ...
  5. def __call__(self, *inputs: Tensor) -> List[Tensor]: ...
  6. ...
  7. class DelayedError:
  8. def __init__(self, msg: AnyStr, num_inputs: int) -> None: ...
  9. def __call__(self, inputs: List[Tensor]) -> List[Tensor]: ...
  10. ...