Corrected R2::identity_map per garys suggestion

applied-material-modeling · Dec 20, 2024 · fdc99ec · fdc99ec
1 parent fbc1d14
commit fdc99ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/neml2/models/IncrementalRate.cxx b/src/neml2/models/IncrementalRate.cxx
@@ -76,7 +76,7 @@ IncrementalRate<T>::set_value(bool out, bool dout_din, bool d2out_din2)
   if (dout_din)
   {
     if (_dv.is_dependent())
-      _dv_dt.d(_dv) = Tensor(torch::eye(_dv.base_storage(), _dv.options()) / dt, dt.batch_dim());
+      _dv_dt.d(_dv) = T::identity_map(_dv.options()) / dt;
     if (_t.is_dependent())
       _dv_dt.d(_t) = -_dv / dt / dt;
     if (_tn.is_dependent())

diff --git a/src/neml2/tensors/R2.cxx b/src/neml2/tensors/R2.cxx
@@ -53,6 +53,6 @@ R2::R2(const Rot & r)
 R4
 R2::identity_map(const torch::TensorOptions & options)
 {
-  return SSR4::identity(options);
+  return torch::eye(9, options).view({3, 3, 3, 3});
 }
 } // namespace neml2