triton.language.histogram

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

计算输入张量在num_bins个bin上的直方图，bin的宽度为1，从0开始。

参数:

• input (Tensor) – 输入张量

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

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

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