_functions.pyi 330 B

1234567891011
  1. from typing import AnyStr, List
  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. def __call__(self, inputs: List[Tensor]) -> List[Tensor]: ...