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

small code clean up.

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