文件 | |
| 文件 | lists/stream_compaction.hpp |
函数 | |
| std::unique_ptr< column > | cudf::lists::apply_boolean_mask (lists_column_view const &input, lists_column_view const &boolean_mask, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref()) |
使用 boolean_mask 布尔列表作为掩码过滤 input 列表列中每一行的元素。 更多... | |
| std::unique_ptr< column > | cudf::lists::distinct (lists_column_view const &input, null_equality nulls_equal=null_equality::EQUAL, nan_equality nans_equal=nan_equality::ALL_EQUAL, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref()) |
| 创建一个新的列表列,其中每个列表不包含重复元素。 更多... | |
| std::unique_ptr<column> cudf::lists::apply_boolean_mask | ( | lists_column_view const & | input, |
| lists_column_view const & | boolean_mask, | ||
| rmm::cuda_stream_view | stream = cudf::get_default_stream(), |
||
| rmm::device_async_resource_ref | mr = cudf::get_current_device_resource_ref() |
||
| ) |
使用 boolean_mask 布尔列表作为掩码过滤 input 列表列中的每个元素。
接收一个 LIST 类型的输入列和一个布尔列表列,该函数会生成一个新的 LIST 列,其类型与 input 相同,其中只有当相应的 boolean_mask 非空且为 true 时,才会从输入行复制每个元素。
例如:
input 和 boolean_mask 必须具有相同的行数。输出列与输入列具有相同的行数。只有当相应的 boolean_mask 元素为 true 时,才会将元素复制到输出行。只有当输入行无效时,输出行才无效。
| cudf::logic_error | 如果 boolean_mask 不是“布尔列表”列 |
| cudf::logic_error | 如果 input 和 boolean_mask 具有不同的行数 |
| input | 要过滤的输入列表列视图 |
| boolean_mask | 用于过滤 input 元素的、可包含 null 值的布尔列表列 |
| stream | 用于设备内存操作和内核启动的 CUDA 流 |
| mr | 用于分配返回表设备内存的设备内存资源 |
input 类型相同的列表列,包含过滤后的列表行| std::unique_ptr<column> cudf::lists::distinct | ( | lists_column_view const & | input, |
| null_equality | nulls_equal = null_equality::EQUAL, |
||
| nan_equality | nans_equal = nan_equality::ALL_EQUAL, |
||
| rmm::cuda_stream_view | stream = cudf::get_default_stream(), |
||
| rmm::device_async_resource_ref | mr = cudf::get_current_device_resource_ref() |
||
| ) |
创建一个新的列表列,其中每个列表不包含重复元素。
给定一个列表列 input,每个列表中的不同元素会被复制到相应的输出列表中。列表的顺序得以保留,但每个列表中元素的顺序不作保证。
示例
| input | 输入列表列 |
| nulls_equal | 指定 null 元素是否应视为相等的标志 |
| nans_equal | 指定浮点 NaN 是否应视为相等的标志 |
| stream | 用于设备内存操作和内核启动的 CUDA 流 |
| mr | 用于分配返回对象的设备内存资源 |