def METHOD_NAME():
    arr = torch.rand(size=[3, 5, 6], device="cuda")
    tensor_list = TensorListGPU(to_dlpack(arr), "NHWC")
    dali_torch_tensor = convert_to_torch(tensor_list, device=arr.device, dtype=arr.dtype)
    assert torch.all(arr.cpu().eq(dali_torch_tensor.cpu()))