提交 080f282d authored 作者: Sigurd Spieckermann's avatar Sigurd Spieckermann

updated docstring

上级 475fbeb5
...@@ -1920,7 +1920,10 @@ def local_set_to_inc_subtensor(node): ...@@ -1920,7 +1920,10 @@ def local_set_to_inc_subtensor(node):
@gof.local_optimizer([Subtensor, AdvancedSubtensor1]) @gof.local_optimizer([Subtensor, AdvancedSubtensor1])
def local_useless_subtensor(node): def local_useless_subtensor(node):
""" """
Remove Subtensor if it takes the full input Remove Subtensor/AdvancedSubtensor1 if it takes the full input. In the
AdvancedSubtensor1 case, the full input is taken when the indices are
equivalent to `arange(0, input.shape[0], 1)` using either an explicit
list/vector or the ARange op.
""" """
# This optimization needs ShapeOpt and fgraph.shape_feature # This optimization needs ShapeOpt and fgraph.shape_feature
if not hasattr(node.fgraph, 'shape_feature'): if not hasattr(node.fgraph, 'shape_feature'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论