提交 264d8bf2 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

BatchedDot: fix reference

上级 914ebf7e
...@@ -3408,8 +3408,7 @@ def batched_dot(a, b): ...@@ -3408,8 +3408,7 @@ def batched_dot(a, b):
a, b, [[a.ndim - 1], [numpy.maximum(1, b.ndim - 2)]]) a, b, [[a.ndim - 1], [numpy.maximum(1, b.ndim - 2)]])
else: else:
# avoid circular import # avoid circular import
from blas import BatchedDot return theano.tensor.blas.BatchedDot()(a, b)
return BatchedDot()(a, b)
def batched_tensordot(x, y, axes=2): def batched_tensordot(x, y, axes=2):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论