pytorch/torch/nn/utils
2022-05-17 05:40:46 +00:00
..
_expanded_weights expanded weights: instance norm faster rule 2022-04-19 19:40:09 +00:00
__init__.py move the stateless util to public API! 2022-04-21 13:42:24 +00:00
_per_sample_grad.py expanded weights without fast rules (#70140) 2022-02-22 20:35:16 +00:00
_stateless.py Added setattr to functional_call. (#77137) 2022-05-17 05:40:46 +00:00
clip_grad.py Update clip_grad_norm_ documentation 2022-04-22 14:07:34 +00:00
convert_parameters.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00
fusion.py [fusion] Add ConvTranspose+BN fusion support (#70022) 2021-12-20 18:42:48 -08:00
init.py Helper function for skipping module parameter / buffer initialization (#57555) 2021-05-24 11:28:32 -07:00
memory_format.py
parametrizations.py Prefer mT and mH over transpose(-2, -1) and transpose(-2, -1).conj() (#64181) 2021-10-18 13:02:25 -07:00
parametrize.py Support for tensor subclasses as parameters 2022-04-27 19:28:55 +00:00
prune.py [lint] upgrade mypy to latest version 2022-05-03 20:51:34 +00:00
rnn.py Convert type comments to annotations in torch/nn (#72662) 2022-02-11 06:35:42 +00:00
rnn.pyi Add pad_sequence as a native function (#57868) 2021-05-11 11:18:13 -07:00
spectral_norm.py Type annotate torch.nn.Module ctor (#61334) 2021-07-16 13:59:06 -07:00
stateless.py Added setattr to functional_call. (#77137) 2022-05-17 05:40:46 +00:00
weight_norm.py torch.nn.modules.LazyModuleMixin and torch.nn.LazyLinear (Shape Inference II) (#44538) 2020-10-19 13:13:54 -07:00