def test_odd(self, device): kernel = _get_center_kernel2d(3, 3).to(device) expected = torch.tensor([[[[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]], [[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]]], [[[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]], [[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]]]]).to(device) assert_allclose(kernel, expected)
def test_odd(self, device, dtype): kernel = _get_center_kernel2d(3, 3, device=device).to(dtype=dtype) expected = torch.tensor([[[[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]], [[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]]], [[[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]], [[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]]]], device=device, dtype=dtype) assert_allclose(kernel, expected, atol=1e-4, rtol=1e-4)
def test_even(self, device): kernel = _get_center_kernel2d(2, 2).to(device) expected = torch.ones(2, 2, 2, 2).to(device) * 0.25 expected[0, 1] = 0 expected[1, 0] = 0 assert_allclose(kernel, expected)
def test_smoke(self, device): kernel = _get_center_kernel2d(3, 4).to(device) assert kernel.shape == (2, 2, 3, 4)
def test_even(self, device, dtype): kernel = _get_center_kernel2d(2, 2, device=device).to(dtype=dtype) expected = torch.ones(2, 2, 2, 2, device=device, dtype=dtype) * 0.25 expected[0, 1] = 0 expected[1, 0] = 0 assert_allclose(kernel, expected, atol=1e-4, rtol=1e-4)