Ejemplo n.º 1
0
 def test_optional(self):
     has_value = cpp_extension.function_taking_optional(torch.ones(5))
     self.assertTrue(has_value)
     has_value = cpp_extension.function_taking_optional(None)
     self.assertFalse(has_value)
Ejemplo n.º 2
0
 def test_optional(self):
     has_value = cpp_extension.function_taking_optional(torch.ones(5))
     self.assertTrue(has_value)
     has_value = cpp_extension.function_taking_optional(None)
     self.assertFalse(has_value)