提交 38558785 authored 作者: Olivier Mastropietro's avatar Olivier Mastropietro

Fixed python3 error using range

上级 a5389ddd
...@@ -203,7 +203,7 @@ def local_dimshuffle_subtensor(node): ...@@ -203,7 +203,7 @@ def local_dimshuffle_subtensor(node):
# the arguments missing from the dimshuffles must be dims that are broadcastable # the arguments missing from the dimshuffles must be dims that are broadcastable
broadcastable = input_.broadcastable broadcastable = input_.broadcastable
missing_dims = range(input_.ndim) missing_dims = list(range(input_.ndim))
for dim in new_order: for dim in new_order:
missing_dims.remove(dim) missing_dims.remove(dim)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论