提交 451ba5fa authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix Crossentropy local track specifier.

上级 152aa23e
......@@ -1417,7 +1417,7 @@ def _is_const(z, val, approx=False):
@opt.register_specialize
@gof.local_optimizer([subtensor.AdvancedSubtensor])
@gof.local_optimizer([subtensor.AdvancedSubtensor, tensor.log])
def local_advanced_indexing_crossentropy_onehot(node):
log = None
sm = None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论