提交 7e7998ec authored 作者: James Bergstra's avatar James Bergstra

c_code for ScalarFromTensor

上级 84f0c415
......@@ -1572,6 +1572,15 @@ class ScalarFromTensor(Op):
return [tensor_from_scalar(dt)]
def __str__(self):
return self.__class__.__name__
def c_code(self, node, name, inputs, outputs, sub):
x, = inputs
z, = outputs
fail = sub['fail']
return """
%(z)s = ((dtype_%(x)s*)(%(x)s->data))[0];
"""%locals()
def c_code_cache_version(self):
return (1,)
scalar_from_tensor = ScalarFromTensor()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论