Python PyTorch histc用法及代码示例

本文简要介绍python语言中 torch.histc 的用法。

用法:

torch.histc(input, bins=100, min=0, max=0, *, out=None) → Tensor

参数：

• input(Tensor) -输入张量。

• bins(int) -直方图箱数

• min(int) -范围的下限(包括)

• max(int) -范围的上限(包括)

关键字参数：

out(Tensor,可选的) -输出张量。

返回：

直方图表示为张量

返回类型：

Tensor

计算张量的直方图。

元素被分类到 min 和 max 之间等宽的 bin 中。如果 min 和 max 都为零，则使用数据的最小值和最大值。

低于 min 和高于 max 的元素将被忽略。

例子：

>>> torch.histc(torch.tensor([1., 2, 1]), bins=4, min=0, max=3)
tensor([ 0.,  2.,  1.,  0.])