提交 75e6dc8d authored 作者: Ying Zhang's avatar Ying Zhang 提交者: Arnaud Bergeron

take1 instead

上级 7f4dd684
...@@ -577,7 +577,7 @@ class GpuAdvancedSubtensor(HideC, tensor.AdvancedSubtensor): ...@@ -577,7 +577,7 @@ class GpuAdvancedSubtensor(HideC, tensor.AdvancedSubtensor):
in zip(items, strides[::-1])], in zip(items, strides[::-1])],
axis=0) axis=0)
# step 6: advanced slicing # step 6: advanced slicing
out_flat = input_flat.take(new_idx.flatten()) out_flat = input_flat.take1(new_idx.flatten())
# step 7: reshape into right shape # step 7: reshape into right shape
out_flat_shp = (x.shape[:start_] + out_flat_shp = (x.shape[:start_] +
new_idx.shape + x.shape[end_:]).astype('int32') new_idx.shape + x.shape[end_:]).astype('int32')
...@@ -588,7 +588,7 @@ class GpuAdvancedSubtensor(HideC, tensor.AdvancedSubtensor): ...@@ -588,7 +588,7 @@ class GpuAdvancedSubtensor(HideC, tensor.AdvancedSubtensor):
out[0] = o.__getitem__(idx_) out[0] = o.__getitem__(idx_)
class GpuAdvancedIncSubtensor1(HideC, tensor.AdvancedIncSubtensor1): class GpuAdvancedIncSubtensor1(Op):
""" """
Implement AdvancedIncSubtensor1 on the gpu. Implement AdvancedIncSubtensor1 on the gpu.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论