Skip to content

P
pytensor
提交 6ecd6c6c authored 作者: Ludwig Schmidt-Hackenberg's avatar Ludwig Schmidt-Hackenberg
浏览文件
操作

Swapped node.inputs[0].data with theano.tensor.extract_constant(node.inputs[0])

See comment by nouiz: https://github.com/shackenberg/Theano/commit/d799afcae312ce973864ea8039b626c862ca93d2#commitcomment-3726133
上级 403ab069
显示空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
theano/sandbox/cuda/basic_ops.py
浏览文件 @ 6ecd6c6c
...@@ -2942,7 +2942,7 @@ class GpuJoin(tensor.Join, GpuOp): ...@@ -2942,7 +2942,7 @@ class GpuJoin(tensor.Join, GpuOp):
out[0] = rval out[0] = rval
def c_code(self, node, name, inputs, out_, sub): def c_code(self, node, name, inputs, out_, sub):
if node.inputs[0].data not in [0, 1]: if theano.tensor.extract_constant(node.inputs[0]) not in [0, 1]:
raise NotImplementedError() raise NotImplementedError()
# only works for the first two axis # only works for the first two axis
if len(inputs) != 3: if len(inputs) != 3:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论