提交 1ca702ef authored 作者: Ian Goodfellow's avatar Ian Goodfellow

improved doc

上级 80130a95
......@@ -493,7 +493,7 @@ class GpuCAReduce(GpuOp):
is a tuple of booleans (actually integers 0 or 1) that specify for
each input dimension, whether to reduce it (1) or not (0).
For example, when scalar_op is theano.scalar.basic.Add:
For example, when scalar_op is a theano.scalar.basic.Add instance:
- reduce_mask == (1,) sums a vector to a scalar
......@@ -506,6 +506,12 @@ class GpuCAReduce(GpuOp):
:note: any reduce_mask of all zeros is a sort of 'copy', and may
be removed during graph optimization
This Op is a work in progress.
This op was recently upgraded from just GpuSum a general CAReduce. Not
many code cases are supported for scalar_op being anything other than
scal.Add instances yet.
"""
def __init__(self, reduce_mask, scalar_op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论