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

take1 instead

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