Function check_layout

dali/test/python/operator_2/test_reduce.py:126–130

(tensor, in_layout, axes, keep_dims)

def check_layout(tensor, in_layout, axes, keep_dims):
    expected_layout = get_expected_layout(in_layout, axes, keep_dims)
    assert (
        tensor.layout() == expected_layout
    ), f"Layout mismatch. Got: `{tensor.layout()}`, expected `{expected_layout}` (axes: {axes})"


def run_dali(