[ 返回参数类型不一致]torch.histogram

torch.histogram

torch.histogram(input, bins, *, range=None, weight=None, density=False, out=None)

paddle.histogram

paddle.histogram(input, bins=100, min=0, max=0, name=None)

PyTorch 相比 Paddle 支持更多其他参数,返回参数 Tensor 数量不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input input 输入 Tensor。
bins bins 直方图 bins(直条)的个数。
range min, max PyTorch 为 bins 的范围,类型为 float,Paddle 为 range 的下边界,上边界,类型为 int,需要转写。
weight - 权重,Paddle 无此参数,暂无转写方式。
density - 结果中每个 bin 是否包含权重数,Paddle 无此参数,暂无转写方式。
返回值 返回值 PyTorch 返回 hist 和 bin_edges,Paddle 返回 hist,暂无转写方式。

转写示例

range 参数:bins 的范围

# PyTorch 写法:
x = torch.tensor([1., 2, 1])
y = torch.histogram(x, bins=5, range=(0., 3.))

# Paddle 写法:
x = paddle.to_tensor([1, 2, 1])
y = paddle.histogram(x, bins=5, min=0, max=3)