(self)
| 3350 | self.assertEqual(Tensor([1.0, 2.0]).masked_select(Tensor([True, False]), size=3, fill_value=-1).dtype, dtypes.default_float) |
| 3351 | |
| 3352 | def test_nonzero(self): |
| 3353 | helper_test_op([(32, 10)], lambda x: (x>0.5).nonzero().int(), lambda x: (x>0.5).nonzero(), forward_only=True) |
| 3354 | helper_test_op([(20,)], lambda x: (x>0.5).nonzero().int(), lambda x: (x>0.5).nonzero(), forward_only=True) |
| 3355 | helper_test_op([(10, 5, 3)], lambda x: (x>0.5).nonzero().int(), lambda x: (x>0.5).nonzero(), forward_only=True) |
| 3356 | for v in (0, 1, 0.0, 2.5, True, False): |
| 3357 | helper_test_op(None, lambda x: x.nonzero().int(), lambda x: x.nonzero(), vals=[v], forward_only=True) |
| 3358 | |
| 3359 | @unittest.skipIf(COMPILE_ONLY, "test requires runtime") |
| 3360 | def test_nonzero_size(self): |
nothing calls this directly
no test coverage detected