Method test_grayscale_torch

tests/test_kornia.py:15–19

(self)

        self.assertEqual(img_t.shape, (3, 1, 1))

    def test_grayscale_torch(self):
        img_rgb = torch.rand(2, 3, 4, 5)
        img_gray = kornia.color.rgb_to_grayscale(img_rgb)

        self.assertEqual(img_gray.shape, (2, 1, 4, 5))