提交 fe15e2f1 authored 作者: Frederic's avatar Frederic

Fix test in the new back-end as in the old back-end

上级 783fc81c
......@@ -427,7 +427,8 @@ class GpuAdvancedIncSubtensor1(HideC, tensor.AdvancedIncSubtensor1):
if len(idx) == 0:
pass
elif y.ndim == x.ndim:
# if len(y) == 1, we need to broadcast it.
elif y.ndim == x.ndim and len(y) != 1:
assert len(y) == len(idx)
k = self.getInplElemwiseAdditionKernel(x[0], y[0])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论