def METHOD_NAME():
    np.random.seed(42)
    for device in ['cpu', 'gpu']:
        for j in range(1, 5):  # test 1..4 levels of nested 'multi-dimensional' scalars
            layout = get_initial_layout(j)
            class RandomScalarLikeTensors():
                def __init__(self, i):
                    self.i = i
                def __call__(self):
                    return random_scalar_like_tensors_batch(self.i)
            for i in range(5):
                yield partial(check_one_hot_operator, axis=None, axis_name='O',
                              expected_output_dim=1,
                              initial_layout=layout), RandomScalarLikeTensors(j), device