提交 5b498c64 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Minor typo fixes

上级 a87ac217
......@@ -1332,9 +1332,9 @@ class CAReduceDtype(CAReduce):
:param scalar_op: a binary scalar op with only one output.
It must be commutative and associative.
:axis: - the dimension along which we want to reduce
- list of dimensions that we want to reduce
- if None, all dimensions are reduced
:param axis: - the dimension along which we want to reduce
- list of dimensions that we want to reduce
- if None, all dimensions are reduced
:param dtype: The dtype of the internal accumulator and returned
tensor. If None, then we use the default dtype which is the same as the
......@@ -1360,7 +1360,7 @@ class CAReduceDtype(CAReduce):
def _output_dtype(self, idtype):
dtype = self.dtype
if dtype is None:
# If input has an discrete dtype, upcast it to 64
# If input has a discrete dtype, upcast it to 64
return dict(
int8='int64',
int16='int64',
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论