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 (triton.int1 的块，可选) – 如果 mask[idx] 为 false，则从直方图中排除 input[idx]

此函数也可以作为 tensor 的成员函数调用，即使用 x.histogram(...) 代替 histogram(x, ...)。