pytorch/torch/nn/modules
2023-12-18 21:29:14 +00:00
..
__init__.py Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
_functions.py [BE]: ruff FURB136: replace ternary with min/max (preview) (#114382) 2023-11-22 22:10:01 +00:00
activation.py Fixed typo in activation.py (#111358) 2023-10-16 20:36:55 +00:00
adaptive.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
batchnorm.py Fix _load_from_state_dict for num_batches_tracked in batchnorm (#115285) 2023-12-07 22:48:26 +00:00
channelshuffle.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
container.py Fix docstring errors in container.py, _functions.py, transformer.py, comm.py, parallel_apply.py, data_parallel.py, scatter_gather.py (#113250) 2023-11-10 21:07:25 +00:00
conv.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
distance.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
dropout.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
flatten.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
fold.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
instancenorm.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
lazy.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
linear.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
loss.py [Doc] Add weight dtype in torch.nn.CrossEntropyLoss (#110998) 2023-10-11 19:52:13 +00:00
module.py improve annotation device parameters where a device ordinal is allowed (#113647) 2023-11-17 14:41:22 +00:00
normalization.py Fix docstrings on torch/nn/modules (#113260) 2023-11-10 18:22:48 +00:00
padding.py [Doc] Add padding size constraint in nn.ReflectionPad2d (#115995) 2023-12-18 21:29:14 +00:00
pixelshuffle.py Issue 112599 - fix pydocstyle errors (#113177) 2023-11-14 20:55:22 +00:00
pooling.py Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
rnn.py Fix RNN class constructor signature (#115341) 2023-12-07 19:46:33 +00:00
sparse.py Issue 112599 - fix pydocstyle errors (#113177) 2023-11-14 20:55:22 +00:00
transformer.py Fix numpy warning when importing torch without numpy installed (#115867) 2023-12-15 02:22:12 +00:00
upsampling.py Issue 112599 - fix pydocstyle errors (#113177) 2023-11-14 20:55:22 +00:00
utils.py Adjust _list_with_default to also work with SymInt input (#113073) 2023-11-07 00:59:25 +00:00