Adding a Checkerboard mesh utility to Pytorch3d

Summary: Adding a checkerboard mesh utility to Pytorch3d.

Reviewed By: bottler

Differential Revision: D39718916

fbshipit-source-id: d43cd30e566b5db068bae6eed0388057634428c8

tests/test_checkerboard.py

@@ -0,0 +1,21 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+import unittest
+
+import torch
+from pytorch3d.utils import checkerboard
+
+from .common_testing import TestCaseMixin
+
+
+class TestCheckerboard(TestCaseMixin, unittest.TestCase):
+    def test_simple(self):
+        board = checkerboard(5)
+        verts = board.verts_packed()
+        expect = torch.tensor([5.0, 5.0, 0])
+        self.assertClose(verts.min(dim=0).values, -expect)
+        self.assertClose(verts.max(dim=0).values, expect)