提交 84da7d76 authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Correct svd op for numpy version 1.12 (changed interface)

上级 519706fa
......@@ -625,9 +625,13 @@ class SVD(Op):
(x,) = inputs
(w, u, v) = outputs
assert x.ndim == 2, "The input of svd function should be a matrix."
w[0], u[0], v[0] = self._numop(x,
self.full_matrices,
self.compute_uv)
if self.compute_uv:
w[0], u[0], v[0] = self._numop(x,
self.full_matrices,
self.compute_uv)
else:
u[0] = self._numop(x, self.full_matrices, self.compute_uv)
def svd(a, full_matrices=1, compute_uv=1):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论