提交 fda86e28 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Documented dtype=None in tensor.mean()

上级 2b9e51f7
...@@ -2686,8 +2686,10 @@ def mean(input, axis=None, dtype=None, op=False): ...@@ -2686,8 +2686,10 @@ def mean(input, axis=None, dtype=None, op=False):
:param dtype: dtype to use for the inner summation. This will not :param dtype: dtype to use for the inner summation. This will not
necessarily be the dtype of the output (in particular necessarily be the dtype of the output (in particular
if it is a discrete (int/uint) dtype, the output will if it is a discrete (int/uint) dtype, the output will
be in a float type) be in a float type).
:type dtype: string If None, then we use float64 for a discrete input, and the
same rules as `sum()` for a continuous input.
:type dtype: None or string
:note: for gpu, if you specify dtype=float32, everything will be done :note: for gpu, if you specify dtype=float32, everything will be done
on the gpu. on the gpu.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论