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)
