[ 仅 paddle 参数更多 ] torch.nansum

torch.nansum

torch.nansum(input, *, dtype=None)

paddle.nansum

paddle.nansum(x, axis=None, dtype=None, keepdim=False, name=None)

Paddle 比 PyTorch 支持更多参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的 Tensor,数据类型为:float16、float32、float64、int32 或 int64。仅参数名不一致。
- axis 求和运算的维度。Pytorch 无此参数,Paddle 保持默认即可。
dtype dtype 输出变量的数据类型。若参数为空,则输出变量的数据类型和输入变量相同,默认值为 None。
- keepdim 是否在输出 Tensor 中保留减小的维度。Pytorch 无此参数,Paddle 保持默认即可。