pytorch/test/nn
mingfeima c6942dbbfb add shape check for random_samples in fractional_max_pool{2d|3d} (#89992)
This PR add shape checks for `random_samples` in fractional_max_pool2d and fractional_max_pool3d.,
to provide more meaningful warnings instead of SegFault when the input is illegal.

For more details, please check https://github.com/pytorch/pytorch/issues/89648
Pull Request resolved: https://github.com/pytorch/pytorch/pull/89992
Approved by: https://github.com/jgong5, https://github.com/ezyang
2022-12-06 14:14:41 +00:00
..
test_convolution.py [flaky] relax tolerance conv1d_vs_scipy (#89193) 2022-11-18 07:31:10 +00:00
test_dropout.py [test_nn] fix missing class attributes for NNTestCase (#89200) 2022-11-22 22:55:44 +00:00
test_embedding.py add _freeze for embedding op (#86769) 2022-10-13 20:12:52 +00:00
test_init.py [test_nn] move init tests from test_nn (#89202) 2022-11-23 08:30:51 +00:00
test_lazy_modules.py Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
test_module_hooks.py [test_nn] split hooks test from test_nn (#89201) 2022-11-23 08:39:45 +00:00
test_multihead_attention.py [test_nn] split multihead_attention from test_nn (#89748) 2022-11-29 18:15:18 +00:00
test_packed_sequence.py
test_parametrization.py [test_nn] split parametrization test from test_nn (#89552) 2022-11-23 17:27:40 +00:00
test_pooling.py add shape check for random_samples in fractional_max_pool{2d|3d} (#89992) 2022-12-06 14:14:41 +00:00
test_pruning.py [test_nn] split pruning tests from test_nn (#89590) 2022-11-24 21:41:22 +00:00