提交 4fea9b5e authored 作者: Joseph Turian's avatar Joseph Turian

Better ValueError in Tensor.dot

上级 867d1252
......@@ -1769,7 +1769,11 @@ class Dot(Op):
return Apply(self, inputs, outputs)
def perform(self, node, (x, y), (z, )):
z[0] = numpy.asarray(numpy.dot(x, y))
try:
z[0] = numpy.asarray(numpy.dot(x, y))
except ValueError, e:
e.args = e.args + (x.shape, y.shape)
raise
def grad(self, (x, y), (gz,)):
if gz.type.ndim == 0:
return gz * y, gz * x
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论