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

torch.histc ¶

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

         

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

         

其中 PyTorch 与 Paddle 的返回值类型不一致，具体如下：

PyTorch	PaddlePaddle	备注
input	input	表示输入的 Tensor。
bins	bins	表示直方图直条的个数。
min	min	表示范围的下边界。
max	max	表示范围的上边界。
out	-	表示输出的 Tensor，Paddle 无此参数，需要转写。
返回值	返回值	表示返回值，PyTorch 的返回值类型为 float32，Paddle 的返回值类型为 int64 ，需要转写。

           # PyTorch 写法
torch.histc(x, out=y)

# Paddle 写法
paddle.assign(paddle.histogram(x).astype('float32'), y)