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

improved doc

上级 80130a95
...@@ -493,7 +493,7 @@ class GpuCAReduce(GpuOp): ...@@ -493,7 +493,7 @@ class GpuCAReduce(GpuOp):
is a tuple of booleans (actually integers 0 or 1) that specify for is a tuple of booleans (actually integers 0 or 1) that specify for
each input dimension, whether to reduce it (1) or not (0). 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 - reduce_mask == (1,) sums a vector to a scalar
...@@ -506,6 +506,12 @@ class GpuCAReduce(GpuOp): ...@@ -506,6 +506,12 @@ class GpuCAReduce(GpuOp):
:note: any reduce_mask of all zeros is a sort of 'copy', and may :note: any reduce_mask of all zeros is a sort of 'copy', and may
be removed during graph optimization 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): def __init__(self, reduce_mask, scalar_op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论