torcheval.metrics.functional.num_collisions¶

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

计算给定输入列表(ids)中的碰撞次数。

Parameters:: 输入 (张量) – 一个输入ID的张量（样本数，）。形状为（样本数，类别数）的类别概率。

示例：

>>> 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])