提交 811c4f88 authored 作者: Colin Raffel's avatar Colin Raffel

CudaNdarray doesn't support list indexing, so convert to tuple first

上级 7a78f453
...@@ -3213,7 +3213,7 @@ class Split(Op): ...@@ -3213,7 +3213,7 @@ class Split(Op):
for i in xrange(self.len_splits): for i in xrange(self.len_splits):
upper_idx = lower_idx + splits[i] upper_idx = lower_idx + splits[i]
general_key[axis] = slice(lower_idx, upper_idx, None) general_key[axis] = slice(lower_idx, upper_idx, None)
outputs[i][0] = x.__getitem__(general_key).copy() outputs[i][0] = x.__getitem__(tuple(general_key)).copy()
lower_idx = upper_idx lower_idx = upper_idx
def infer_shape(self, node, in_shapes): def infer_shape(self, node, in_shapes):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论