提交 11b87f2d authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Minor updates to inline doc of rebroadcast &co.

上级 93a65c3d
...@@ -3096,7 +3096,7 @@ class Rebroadcast(Op): ...@@ -3096,7 +3096,7 @@ class Rebroadcast(Op):
e.g.: Rebroadcast((0, True), (1, False))(x) e.g.: Rebroadcast((0, True), (1, False))(x)
would make x broadcastable in axis 0 would make x broadcastable in axis 0
and not broadcastable in axis 1 and not broadcastable in axis 1
See also the unbroadcast function. See also the unbroadcast, addbroadcast and patternbroadcast functions.
..note: work inplace and work for CudaNdarrayType ..note: work inplace and work for CudaNdarrayType
""" """
...@@ -3137,7 +3137,7 @@ def addbroadcast(x, *axes): ...@@ -3137,7 +3137,7 @@ def addbroadcast(x, *axes):
""" """
Make the input broadcastable in the specified axes. Make the input broadcastable in the specified axes.
We apply the opt here to don't pollute the graph especially during the gpu optimization We apply the opt here not to pollute the graph especially during the gpu optimization
""" """
rval = Rebroadcast(*[(axis, True) for axis in axes])(x) rval = Rebroadcast(*[(axis, True) for axis in axes])(x)
return theano.tensor.opt.apply_rebroadcast_opt(rval) return theano.tensor.opt.apply_rebroadcast_opt(rval)
...@@ -3146,16 +3146,16 @@ def unbroadcast(x, *axes): ...@@ -3146,16 +3146,16 @@ def unbroadcast(x, *axes):
""" """
Make the input impossible to broadcast in the specified axes. Make the input impossible to broadcast in the specified axes.
We apply the opt here to don't pollute the graph especially during the gpu optimization We apply the opt here not to pollute the graph especially during the gpu optimization
""" """
rval = Rebroadcast(*[(axis, False) for axis in axes])(x) rval = Rebroadcast(*[(axis, False) for axis in axes])(x)
return theano.tensor.opt.apply_rebroadcast_opt(rval) return theano.tensor.opt.apply_rebroadcast_opt(rval)
def patternbroadcast(x, broadcastable): def patternbroadcast(x, broadcastable):
""" """
Make the input impossible to broadcast in the specified axes. Make the input adopt a specific broadcasting pattern.
We apply the opt here to don't pollute the graph especially during the gpu optimization We apply the opt here not to pollute the graph especially during the gpu optimization
""" """
rval = Rebroadcast(*[(i,broadcastable[i]) for i in range(len(broadcastable))])(x) rval = Rebroadcast(*[(i,broadcastable[i]) for i in range(len(broadcastable))])(x)
return theano.tensor.opt.apply_rebroadcast_opt(rval) return theano.tensor.opt.apply_rebroadcast_opt(rval)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论