torcheval.metrics.functional.num_collisions

torcheval.metrics.functional.num_collisions(input: Tensor) → Tensor

给定输入（id）列表，计算碰撞次数。

参数：:

input (Tensor) – 输入 id 张量 (num_samples, )。形状为 (num_samples, num_classes) 的类别概率。

示例

>>> import torch
>>> from torcheval.metrics.functional import num_collisions
>>> input = torch.tensor([3, 4, 2, 3])
>>> num_collisions(input)
tensor([1, 0, 0, 1])
>>> input = torch.tensor([3, 4, 1, 3, 1, 1, 5])
>>> num_collisions(input)
tensor([1, 0, 2, 1, 2, 2, 0])