提交 3e5f365f authored 作者: James Bergstra's avatar James Bergstra

Added support for a scalar special-case in GpuElemwise

上级 f55c4b65
...@@ -641,7 +641,7 @@ class NaiveAlgo(object): ...@@ -641,7 +641,7 @@ class NaiveAlgo(object):
output_args = ", ".join("o%i_data, o%i_str"%(ipos, ipos) output_args = ", ".join("o%i_data, o%i_str"%(ipos, ipos)
for ipos in xrange(len(node.outputs))) for ipos in xrange(len(node.outputs)))
prod_dims = '*'.join("dims[%i]"%di for di in xrange(nd)) prod_dims = '*'.join(["dims[%i]"%di for di in xrange(nd)]+['1'])
scalar_op=self.scalar_op.__class__.__name__ scalar_op=self.scalar_op.__class__.__name__
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论