快捷方式

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源