pytorch/torch/utils/hooks.pyi
Jon Malmaud bfeff1eb8f Stubs for torch.nn (#19089)
Summary:
Closes https://github.com/pytorch/pytorch/issues/18724
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19089

Differential Revision: D16073654

Pulled By: ezyang

fbshipit-source-id: 5642179651ce45ab7c5a46cc1fcc4fd6b37fa71c
2019-07-01 09:50:17 -07:00

9 lines
232 B
Python

from typing import Any
class RemovableHandle:
def __init__(self, hooks_dict: Any) -> None: ...
def remove(self) -> None: ...
def __enter__(self): ...
def __exit__(self, type: Any, value: Any, tb: Any) -> None: ...