torcharrow.functional.get_score_min¶
- torcharrow.functional.get_score_min(input_ids: ListColumn, matching_ids: ListColumn, matching_id_scores: ListColumn)¶
返回 matching_id_scores 中所有分数的最小值,这些分数对应的 id 在 matching_ids 中,并且也在 input_ids 中。
- 参数:
input_ids (第一个 id 列表) –
matching_ids (第二个 id 列表) –
matching_ids_scores (matching_ids 的分数(权重)) –
示例
>>> import torcharrow as ta >>> from torcharrow import functional >>> input_ids = ta.column([[1, 2, 3]]) >>> matching_ids = ta.column([[1,2]]) >>> matching_id_scores = ta.column([[5.0,4.0]]) >>> functional.get_score_min(input_ids, matching_ids, matching_id_scores) 0 4.0 dtype: Float32(nullable=True), length: 1, null_count: 0