提交 43eec188 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

We don't need the inputs to be contiguous.

上级 3d6798fb
...@@ -405,7 +405,7 @@ class GpuAdvancedSubtensor1(HideC, tensor.AdvancedSubtensor1): ...@@ -405,7 +405,7 @@ class GpuAdvancedSubtensor1(HideC, tensor.AdvancedSubtensor1):
""" """
def make_node(self, x, ilist): def make_node(self, x, ilist):
ctx_name = infer_context_name(x, ilist) ctx_name = infer_context_name(x, ilist)
x_ = gpu_contiguous(as_gpuarray_variable(x, ctx_name)) x_ = as_gpuarray_variable(x, ctx_name)
ilist__ = tensor.as_tensor_variable(ilist) ilist__ = tensor.as_tensor_variable(ilist)
if ilist__.type.dtype not in tensor.integer_dtypes: if ilist__.type.dtype not in tensor.integer_dtypes:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论