提交 59fa3b3f authored 作者: Frederic's avatar Frederic

small code clean up.

上级 e1c2bd9e
......@@ -300,7 +300,7 @@ class GpuDimShuffle(GpuOp):
for i, b in enumerate(input_broadcastable):
if i not in new_order:
if b != 1:
if not b:
# we cannot drop non-broadcastable dimensions
raise ValueError("You cannot drop a non-broadcastable"
" dimension.",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论