1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
# This file is never automatically imported within PyTorch so it is ok to
# always warn here
import warnings
warnings.warn("The `torch.nn.utils._stateless` code is deprecated now that "
"it is publicly available. Please use `torch.nn.utils.stateless "
"instead.", DeprecationWarning)
# Import * wouldn't work as most things are private and thus wouldn't be imported
# here.
from torch.nn.utils.stateless import functional_call # noqa: F401
from torch.nn.utils.stateless import _apply_func_submodules, _change_class # noqa: F401
# This one used to look public but should actually be private. This was fixed when making the module
# public and is kept here for BC
from torch.nn.utils.stateless import _reparametrize_module as reparametrize_module # noqa: F401
|
