Skip to content

Commit

Permalink
[SystemDS-3755] Python API Builtin inv
Browse files Browse the repository at this point in the history
  • Loading branch information
e-strauss authored and Baunsgaard committed Sep 4, 2024
1 parent d6fe87e commit f190880
Show file tree
Hide file tree
Showing 2 changed files with 59 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/main/python/systemds/operator/nodes/matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -687,5 +687,12 @@ def replace(
},
)

def inv(self) -> "Matrix":
""" Computes the inverse of a squared matrix.
:return: The Matrix representing the result of this operation
"""
return Matrix(self.sds_context, "inv", [self])

def __str__(self):
return "MatrixNode"
52 changes: 52 additions & 0 deletions src/main/python/tests/matrix/test_inv.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
# -------------------------------------------------------------
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#
# -------------------------------------------------------------

import unittest
import numpy as np
from systemds.context import SystemDSContext

np.random.seed(7)
m = np.random.random((10, 10))


class TestINV(unittest.TestCase):
def setUp(self):
self.sds = SystemDSContext()

def tearDown(self):
self.sds.close()

def test_diag_basic(self):
input_matrix = np.array([[2, 0], [0, 6]])
sds_input = self.sds.from_numpy(input_matrix)
sds_result = sds_input.inv().compute()
np_result = np.linalg.inv(input_matrix)
assert np.allclose(sds_result, np_result, 1e-9)

def test_diag_random(self):
sds_input = self.sds.from_numpy(m)
sds_result = sds_input.inv().compute()
np_result = np.linalg.inv(m)
assert np.allclose(sds_result, np_result, 1e-9)


if __name__ == "__main__":
unittest.main()

0 comments on commit f190880

Please sign in to comment.