提交 4c638396 authored 作者: Dustin Webb's avatar Dustin Webb 提交者: Frederic

Fixed return statement.

上级 1d8302e5
...@@ -1767,7 +1767,7 @@ def local_advanced_indexing_crossentropy_onehot_grad(node): ...@@ -1767,7 +1767,7 @@ def local_advanced_indexing_crossentropy_onehot_grad(node):
if labels.ndim == 1 and x_var.ndim == 2: if labels.ndim == 1 and x_var.ndim == 2:
return [crossentropy_softmax_1hot_with_bias_dx(out_grad, sm, labels)] return [crossentropy_softmax_1hot_with_bias_dx(out_grad, sm, labels)]
else: else:
retur return
@opt.register_specialize('gpu') @opt.register_specialize('gpu')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论