torch.histc
- torch.histc(input, bins=100, min=0, max=0, *, out=None) → Tensor
-
计算张量的直方图。
元素会被分类到
min
和max
之间的等宽区间内。如果min
和max
都为零,则使用数据的最小值和最大值。低于 min 或高于 max 的元素,以及
NaN
元素将被忽略。- 参数
- 关键字参数
-
out (Tensor, 可选) – 指定输出张量。
- 返回值
-
以张量形式表示的直方图
- 返回类型
示例:
>>> torch.histc(torch.tensor([1., 2, 1]), bins=4, min=0, max=3) tensor([ 0., 2., 1., 0.])