后处理函数

模块：polygraphy.comparator

class PostprocessFunc

基类：object

提供可以将后处理应用于 IterationResult 的函数。

static top_k(k=None)

创建一个函数，该函数将 Top-K 操作应用于 IterationResult。 Top-K 将返回数组中 k 个最大值的索引。

参数:

k (Union[int, Tuple[int, int], Dict[str, int], Dict[str, Tuple[int, int]]]) –

要保留的索引数，以及可选的操作轴。例如，(5, 0) 的值将保留沿轴 0 的前 5 个索引。

如果此值超过轴长度，则会被钳制。可以通过提供字典在每个输出的基础上指定此值。在这种情况下，使用空字符串 (“”) 作为键来为未明确列出的输出指定默认的 top-k 值。如果不存在默认值，则未指定的输出将不会被修改。默认为 10。

返回:

top-k 函数。

返回类型:

Callable(IterationResult) -> IterationResult