提交 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):
:param dtype: dtype to use for the inner summation. This will not
necessarily be the dtype of the output (in particular
if it is a discrete (int/uint) dtype, the output will
be in a float type)
:type dtype: string
be in a float type).
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
on the gpu.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论