提交 6fd7f9bf authored 作者: Frederic Bastien's avatar Frederic Bastien

add string representation of CudaNdarrayConstant.

上级 1b4e0a49
...@@ -40,6 +40,10 @@ class CudaNdarrayConstantSignature(tensor.TensorConstantSignature): ...@@ -40,6 +40,10 @@ class CudaNdarrayConstantSignature(tensor.TensorConstantSignature):
class CudaNdarrayConstant(Constant, _operators): class CudaNdarrayConstant(Constant, _operators):
def signature(self): def signature(self):
return CudaNdarrayConstantSignature((self.type, numpy.asarray(self.data))) return CudaNdarrayConstantSignature((self.type, numpy.asarray(self.data)))
def __str__(self):
if self.name is not None:
return self.name
return "CudaNdarray{"+str(numpy.asarray(self.data))+"}"
CudaNdarrayType.Constant = CudaNdarrayConstant CudaNdarrayType.Constant = CudaNdarrayConstant
class CudaNdarraySharedVariable(SharedVariable, _operators): class CudaNdarraySharedVariable(SharedVariable, _operators):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论