torcheval.metrics.functional.frequency_at_k¶

torcheval.metrics.functional.frequency_at_k(input: Tensor, k: float) → Tensor¶

给定频率列表和阈值k，计算频率。生成一个二进制列表以指示频率是否小于k。

Parameters:

input (Tensor) – 预测的未归一化分数（通常称为logits）。
k (float) – 频率的阈值。k 不应为负值。

示例

>>> import torch
>>> from torcheval.metrics.functional import frequency
>>> input = torch.tensor([0.3, 0.1, 0.6])
>>> frequency(input, k=0.5)
tensor([1.0000, 1.0000, 0.0000])