torcharrow.functional.get_score_sum

torcharrow.functional.get_score_sum(input_ids: ListColumn, input_id_scores: ListColumn, matching_ids: ListColumn, matching_id_scores: ListColumn)

傳回 matching_id_scores 中所有分數的總和，這些分數在 matching_ids 中具有對應的 ID，且該 ID 也存在於 input_ids 中。 :param input_ids: :type input_ids: 第一個 ID 列表 :param input_ids_scores: :type input_ids_scores: input_ids 的分數（權重） :param matching_ids: :type matching_ids: 第二個 ID 列表 :param matching_ids_scores: :type matching_ids_scores: matching_ids 的分數（權重）

範例

>>> import torcharrow as ta
>>> from torcharrow import functional
>>> input_ids = ta.column([[1, 2, 3]])
>>> input_id_scores = ta.column([[1.0,2.0,3.0]])
>>> matching_ids = ta.column([[1,2]])
>>> matching_id_scores = ta.column([[5.0,4.0]])
>>> functional.get_score_sum(input_ids, input_id_scores, matching_ids, matching_id_scores)
0  9
dtype: Float32(nullable=True), length: 1, null_count: 0