def METHOD_NAME():
    import torch
    samples_cpu = [
        (torch.tensor(test_array), test_array),
    ]
    batches_cpu = [
        ([torch.tensor(test_array)], [test_array]),
        ([torch.tensor(test_array)] * 4, [test_array] * 4),
    ]
    disallowed_samples = [
        torch.tensor(test_array).cuda(),
    ]
    yield from run_checks(samples_cpu, batches_cpu, disallowed_samples, [])