提交 b72de011 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Small bugfix

上级 673a30b3
...@@ -940,7 +940,7 @@ def local_advanced_indexing_crossentropy_onehot(node): ...@@ -940,7 +940,7 @@ def local_advanced_indexing_crossentropy_onehot(node):
pass pass
if sm is not None and sm.owner and sm.owner.op in (softmax, softmax_with_bias): if sm is not None and sm.owner and sm.owner.op in (softmax, softmax_with_bias):
sm_w_bias = local_softmax_with_bias.transform(sm) sm_w_bias = local_softmax_with_bias.transform(sm.owner)
if sm_w_bias: if sm_w_bias:
assert sm_w_bias[0].owner.op == softmax_with_bias assert sm_w_bias[0].owner.op == softmax_with_bias
x_var, b_var = sm_w_bias[0].owner.inputs x_var, b_var = sm_w_bias[0].owner.inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论