Function _replace_nan

python/paddle/tensor/stat.py:283–288

(out)

    out_tensor /= corrected_n

    def _replace_nan(out):
        indices = paddle.arange(out.numel(), dtype='int64')
        out_nan = paddle.index_fill(
            out.flatten(), indices, 0, float('nan')
        ).reshape(out.shape)
        return out_nan

    if 0 in x.shape:
        out_tensor = _replace_nan(out_tensor)