triton.language.histogram

triton.language.histogram(input, num_bins, mask=None, _semantic=None, _generator=None)

根据输入张量计算一个直方图，该直方图有 num_bins 个箱，箱宽为 1 并从 0 开始。

参数:

• input (Tensor) – 输入张量

• num_bins (int) – 直方图箱的数量

• mask (Block of triton.int1, 可选) – 如果 mask[idx] 为 false，则从直方图中排除 input[idx]

此函数也可以作为成员函数在 tensor 上调用，例如 x.histogram(...) 而不是 histogram(x, ...)。