提交 849c3b86 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Use boolean __props__ in SVD

Fixes failure in more recent versions of jaxlib
上级 931297fc
...@@ -541,9 +541,9 @@ class SVD(Op): ...@@ -541,9 +541,9 @@ class SVD(Op):
# See doc in the docstring of the function just after this class. # See doc in the docstring of the function just after this class.
__props__ = ("full_matrices", "compute_uv") __props__ = ("full_matrices", "compute_uv")
def __init__(self, full_matrices=True, compute_uv=True): def __init__(self, full_matrices: bool = True, compute_uv: bool = True):
self.full_matrices = full_matrices self.full_matrices = bool(full_matrices)
self.compute_uv = compute_uv self.compute_uv = bool(compute_uv)
def make_node(self, x): def make_node(self, x):
x = as_tensor_variable(x) x = as_tensor_variable(x)
...@@ -584,7 +584,7 @@ class SVD(Op): ...@@ -584,7 +584,7 @@ class SVD(Op):
return [s_shape] return [s_shape]
def svd(a, full_matrices=1, compute_uv=1): def svd(a, full_matrices: bool = True, compute_uv: bool = True):
""" """
This function performs the SVD on CPU. This function performs the SVD on CPU.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论