Function make_torch_tensor

dali/test/python/test_external_source_impl.py:94–96

(*args, **kwargs)

    global random_array

    def make_torch_tensor(*args, **kwargs):
        t = torch.tensor(*args, **kwargs)
        return t.cuda() if gpu else t

    def torch_cast(x, dtype):
        return x.type(dtype)