提交 59f09d09 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Allow NumPy dtype objects in MakeVector

上级 6b6ed43a
...@@ -1607,7 +1607,7 @@ class MakeVector(COp): ...@@ -1607,7 +1607,7 @@ class MakeVector(COp):
__props__ = ("dtype",) __props__ = ("dtype",)
def __init__(self, dtype="int64"): def __init__(self, dtype="int64"):
self.dtype = dtype self.dtype = np.dtype(dtype).name
def make_node(self, *inputs): def make_node(self, *inputs):
inputs = list(map(as_tensor_variable, inputs)) inputs = list(map(as_tensor_variable, inputs))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论