コード例 #1
0
def _to_tensor(gdf: cudf.DataFrame, dtype, to_cpu=False):
    if gdf.empty:
        return
    g = gdf.to_dlpack()
    cols = gdf.columns
    t = from_dlpack(g).type(dtype)
    del g, gdf
    return t, cols