未验证 提交 3a301780 编写于 作者: A Asthestarsfalll 提交者: GitHub

update document of quantile and nanquantile; test=document_fix (#42413)

上级 c905a9e9
...@@ -440,8 +440,7 @@ def _compute_quantile(x, q, axis=None, keepdim=False, ignore_nan=False): ...@@ -440,8 +440,7 @@ def _compute_quantile(x, q, axis=None, keepdim=False, ignore_nan=False):
Compute the quantile of the input along the specified axis. Compute the quantile of the input along the specified axis.
Args: Args:
Args: x (Tensor): The input Tensor, it's data type can be float32, float64, int32, int64.
x (Tensor): The input Tensor, it's data type can be float32, float64.
q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list, q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list,
each q will be calculated and the first dimension of output is same to the number of ``q`` . each q will be calculated and the first dimension of output is same to the number of ``q`` .
axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int. axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int.
...@@ -525,7 +524,7 @@ def _compute_quantile(x, q, axis=None, keepdim=False, ignore_nan=False): ...@@ -525,7 +524,7 @@ def _compute_quantile(x, q, axis=None, keepdim=False, ignore_nan=False):
if ignore_nan: if ignore_nan:
indices.append(q_num * (valid_counts - 1)) indices.append(q_num * (valid_counts - 1))
else: else:
# TODO(Asthestarsfalll): Use paddle.index_fill instead of where # TODO: Use paddle.index_fill instead of where
index = q_num * (valid_counts - 1) index = q_num * (valid_counts - 1)
last_index = x.shape[axis] - 1 last_index = x.shape[axis] - 1
nums = paddle.full_like(index, fill_value=last_index) nums = paddle.full_like(index, fill_value=last_index)
...@@ -569,7 +568,7 @@ def quantile(x, q, axis=None, keepdim=False): ...@@ -569,7 +568,7 @@ def quantile(x, q, axis=None, keepdim=False):
If any values in a reduced row are NaN, then the quantiles for that reduction will be NaN. If any values in a reduced row are NaN, then the quantiles for that reduction will be NaN.
Args: Args:
x (Tensor): The input Tensor, it's data type can be float32, float64. x (Tensor): The input Tensor, it's data type can be float32, float64, int32, int64.
q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list, q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list,
each q will be calculated and the first dimension of output is same to the number of ``q`` . each q will be calculated and the first dimension of output is same to the number of ``q`` .
axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int. axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int.
...@@ -629,7 +628,7 @@ def nanquantile(x, q, axis=None, keepdim=False): ...@@ -629,7 +628,7 @@ def nanquantile(x, q, axis=None, keepdim=False):
If all values in a reduced row are NaN, then the quantiles for that reduction will be NaN. If all values in a reduced row are NaN, then the quantiles for that reduction will be NaN.
Args: Args:
x (Tensor): The input Tensor, it's data type can be float32, float64. x (Tensor): The input Tensor, it's data type can be float32, float64, int32, int64.
q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list, q (int|float|list): The q for calculate quantile, which should be in range [0, 1]. If q is a list,
each q will be calculated and the first dimension of output is same to the number of ``q`` . each q will be calculated and the first dimension of output is same to the number of ``q`` .
axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int. axis (int|list, optional): The axis along which to calculate quantile. ``axis`` should be int or list of int.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册