提交 b932c79c authored 作者: Frederic Bastien's avatar Frederic Bastien

In make_node, always call as_tensor_variable. (not a problem most of the time as…

In make_node, always call as_tensor_variable. (not a problem most of the time as not used directly by user.
上级 fd8e0a88
...@@ -1532,6 +1532,8 @@ class Dot22(GemmRelated): ...@@ -1532,6 +1532,8 @@ class Dot22(GemmRelated):
check_input = False check_input = False
def make_node(self, x, y): def make_node(self, x, y):
x = T.as_tensor_variable(x)
y = T.as_tensor_variable(y)
dtypes = ('float16', 'float32', 'float64', 'complex64', 'complex128') dtypes = ('float16', 'float32', 'float64', 'complex64', 'complex128')
if x.type.ndim != 2 or x.type.dtype not in dtypes: if x.type.ndim != 2 or x.type.dtype not in dtypes:
raise TypeError(x) raise TypeError(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论