Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #include <cstddef>
       7              : 
       8              : #include "DataStructures/Tensor/Tensor.hpp"
       9              : #include "Utilities/MakeWithValue.hpp"
      10              : 
      11              : /// \ingroup TensorGroup
      12              : /// \brief returns the Identity matrix
      13              : template <size_t Dim, typename DataType>
      14            1 : tnsr::Ij<DataType, Dim, Frame::NoFrame> identity(
      15              :     const DataType& used_for_type) {
      16              :   auto identity_matrix{make_with_value<tnsr::Ij<DataType, Dim, Frame::NoFrame>>(
      17              :       used_for_type, 0.0)};
      18              : 
      19              :   for (size_t i = 0; i < Dim; ++i) {
      20              :     identity_matrix.get(i, i) = 1.0;
      21              :   }
      22              :   return identity_matrix;
      23              : }