quaterion.eval.accumulators.group_accumulator 模块¶ class GroupAccumulator[来源]¶ 基类: Accumulator 为基于组的任务累积嵌入和组。 reset()[来源]¶ 重置累加器状态 重置累加器状态、累积的嵌入和组 update(embeddings: Tensor, groups: Tensor, device=None)[来源]¶ 更新累加器状态。 将提供的嵌入和组移动到合适的设备并添加到累积状态。 参数: embeddings – 要累积的嵌入 groups – 要累积的相应组 device – 存储张量的设备 property groups¶ 将组列表连接到张量 帮助避免在累积期间对每个批次连接组。相反，只在调用时连接它。 返回: torch.Tensor – 组批次 property state: Dict[str, Tensor]¶ 累积状态 返回: Dict[str, torch.Tensor] - 包含嵌入和组的字典。
