Function make_sample_tensor

dali/test/python/torchvision/test_tv_normalize.py:34–42

()

def make_sample_tensor():
    # 3x2x2 (C, H, W)
    return torch.tensor(
        [
            [[1.0, 2.0], [3.0, 4.0]],
            [[5.0, 6.0], [7.0, 8.0]],
            [[9.0, 10.0], [11.0, 12.0]],
        ]
    )


def test_normalize_core(