提交 14432317 authored 作者: Shawn Tan's avatar Shawn Tan

Corrected `range()` call for python3

上级 97bb9197
...@@ -6572,7 +6572,7 @@ class AllocDiag(Op): ...@@ -6572,7 +6572,7 @@ class AllocDiag(Op):
result[diagonal_slice] = x result[diagonal_slice] = x
if len(x.shape) > 1: if len(x.shape) > 1:
# Re-order axes so they correspond to diagonals at axis1, axis2 # Re-order axes so they correspond to diagonals at axis1, axis2
axes = range(len(x.shape[:-1])) axes = list(range(len(x.shape[:-1])))
last_idx = axes[-1] last_idx = axes[-1]
axes = axes[:axis1] + [last_idx + 1] + axes[axis1:] axes = axes[:axis1] + [last_idx + 1] + axes[axis1:]
axes = axes[:axis2] + [last_idx + 2] + axes[axis2:] axes = axes[:axis2] + [last_idx + 2] + axes[axis2:]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论