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