Method func

tests/structures/test_boxes.py:184–186

(x)

    @unittest.skipIf(TORCH_VERSION < (1, 6), "Insufficient pytorch version")
    def test_scriptability(self):
        def func(x):
            boxes = Boxes(x)
            return boxes.area()

        f = torch.jit.script(func)
        f(torch.rand((3, 4)))