Home
last modified time | relevance | path

Searched refs:_test_multinomial_invalid_probs (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_torch.py9495 def _test_multinomial_invalid_probs(probs): function
9503 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., -1., 1.]))
9504 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., inf, 1.]))
9505 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., -inf, 1.]))
9506 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., 1., nan]))