sleipnirgroup-jormungandr 0.6.5.dev18__tar.gz → 0.6.5.dev20__tar.gz

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sleipnirgroup-jormungandr
-Version: 0.6.5.dev18
+Version: 0.6.5.dev20
 Summary: Reverse mode autodiff library and NLP solver DSL
 License-Expression: BSD-3-Clause
 License-File: LICENSE.txt

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/include/sleipnir/autodiff/variable_matrix.hpp

@@ -1187,6 +1187,22 @@ class VariableMatrix : public SleipnirBase {
   /// @return Number of elements in matrix.
   size_t size() const { return m_storage.size(); }
 
+  /// Returns an identity variable matrix.
+  ///
+  /// @param rows The number of matrix rows.
+  /// @return An identity variable matrix.
+  static VariableMatrix<Scalar> identity(int rows) {
+    VariableMatrix<Scalar> result{detail::empty, rows, rows};
+
+    for (int row = 0; row < rows; ++row) {
+      for (int col = 0; col < rows; ++col) {
+        result[row, col] = row == col ? Scalar(1) : Scalar(0);
+      }
+    }
+
+    return result;
+  }
+
   /// Returns a variable matrix filled with zeros.
   ///
   /// @param rows The number of matrix rows.

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/python/cpp/autodiff/bind_variable_matrix.cpp

@@ -439,6 +439,8 @@ void bind_variable_matrix(nb::module_& autodiff,
          const std::function<Variable<double>(const Variable<double>& x)>&
              unary_op) { return self.cwise_transform(unary_op); },
       "func"_a, DOC(slp, VariableMatrix, cwise_transform));
+  cls.def_static("identity", &VariableMatrix<double>::identity, "rows"_a,
+                 DOC(slp, VariableMatrix, identity));
   cls.def_static("zero", &VariableMatrix<double>::zero, "rows"_a, "cols"_a,
                  DOC(slp, VariableMatrix, zero));
   cls.def_static("one", &VariableMatrix<double>::one, "rows"_a, "cols"_a,

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/python/cpp/docstrings.hpp

@@ -1812,6 +1812,15 @@ R"doc(Returns const end iterator.
 Returns:
     Const end iterator.)doc";
 
+static const char *__doc_slp_VariableMatrix_identity =
+R"doc(Returns an identity variable matrix.
+
+Parameter ``rows``:
+    The number of matrix rows.
+
+Returns:
+    An identity variable matrix.)doc";
+
 static const char *__doc_slp_VariableMatrix_iterator = R"doc()doc";
 
 static const char *__doc_slp_VariableMatrix_iterator_iterator = R"doc()doc";

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/python/src/sleipnir/__init__.py

@@ -2,4 +2,4 @@
 
 from ._sleipnir import *
 
-__version__ = "0.6.5.dev18"
+__version__ = "0.6.5.dev20"

{sleipnirgroup_jormungandr-0.6.5.dev18 → sleipnirgroup_jormungandr-0.6.5.dev20}/python/test/autodiff/variable_matrix_test.py

@@ -358,9 +358,19 @@ def test_unary_broadcast():
     assert (result2.value() == expected2).all()
 
 
+def test_identity_static_function():
+    A = VariableMatrix.identity(3)
+
+    assert A.shape == (3, 3)
+
+    assert (A.value() == np.identity(3)).all()
+
+
 def test_zero_static_function():
     A = VariableMatrix.zero(2, 3)
 
+    assert A.shape == (2, 3)
+
     for row in range(A.rows()):
         for col in range(A.cols()):
             assert A[row, col].value() == 0.0
@@ -369,6 +379,8 @@ def test_zero_static_function():
 def test_one_static_function():
     A = VariableMatrix.one(2, 3)
 
+    assert A.shape == (2, 3)
+
     for row in range(A.rows()):
         for col in range(A.cols()):
             assert A[row, col].value() == 1.0
@@ -377,6 +389,8 @@ def test_one_static_function():
 def test_constant_static_function():
     A = VariableMatrix.constant(2, 3, 2.0)
 
+    assert A.shape == (2, 3)
+
     for row in range(A.rows()):
         for col in range(A.cols()):
             assert A[row, col].value() == 2.0