stages.text.embedders.utils#
Module Contents#
Functions#
Create a cudf list series from 2d arrays. This code comes from https://github.com/rapidsai/crossfit/blob/76f74d0d927cf76313a3960d7dd5575d1dff2f06/crossfit/backend/cudf/series.py#L20-L32 |
API#
- stages.text.embedders.utils.create_list_series_from_1d_or_2d_ar(
- ar: Any,
- index: cudf.Index,
Create a cudf list series from 2d arrays. This code comes from https://github.com/rapidsai/crossfit/blob/76f74d0d927cf76313a3960d7dd5575d1dff2f06/crossfit/backend/cudf/series.py#L20-L32
Args: ar (cp.ndarray): any object that can be converted to a cupy array (cupy, numpy, torch, etc.) index (cudf.Index): index of the the dataframe to be returned
Returns: cudf.Series: cudf series with the index respected