4. hermitian_matrix_rank_2k_update

最終更新日時:
が更新

履歴 編集

function template
<linalg>

std::linalg::hermitian_matrix_rank_2k_update(C++26)

namespace std::linalg {
  template<in-matrix InMat1,
           in-matrix InMat2,
           possibly-packed-out-matrix OutMat,
           class Triangle>
  void hermitian_matrix_rank_2k_update(
    InMat1 A,
    InMat2 B,
    OutMat C,
    Triangle t); // (1)

  template<class ExecutionPolicy,
           in-matrix InMat1,
           in-matrix InMat2,
           possibly-packed-out-matrix OutMat,
           class Triangle>
  void hermitian_matrix_rank_2k_update(
    ExecutionPolicy&& exec,
    InMat1 A,
    InMat2 B,
    OutMat C,
    Triangle t); // (2)

  template<in-matrix InMat1,
           in-matrix InMat2,
           in-matrix InMat3,
           possibly-packed-out-matrix OutMat,
           class Triangle>
  void hermitian_matrix_rank_2k_update(
    InMat1 A,
    InMat2 B,
    InMat3 E,
    OutMat C,
    Triangle t); // (3)

  template<class ExecutionPolicy,
           in-matrix InMat1,
           in-matrix InMat2,
           in-matrix InMat3,
           possibly-packed-out-matrix OutMat,
           class Triangle>
  void hermitian_matrix_rank_2k_update(
    ExecutionPolicy&& exec,
    InMat1 A,
    InMat2 B,
    InMat3 E,
    OutMat C,
    Triangle t); // (4)
}

概要

エルミートな(対称かつ共役を取る)rank-2k updateを対称行列に行う。

  • (1), (2)は上書き(overwriting)版であり、出力行列Cを計算結果で上書きする。
  • (3), (4)は更新(updating)版であり、入力行列Eに更新項を加えた結果を出力行列Cに書き込む。

引数tは対称行列の成分が上三角にあるのか、それとも下三角にあるのかを示す。

  • (1): $C = AB^* + BA^*$
  • (2): (1)を指定された実行ポリシーで実行する。
  • (3): $C = E + AB^* + BA^*$
  • (4): (3)を指定された実行ポリシーで実行する。

適格要件

事前条件

効果

  • (1), (2): $C = AB^* + BA^*$
  • (3), (4): $C = E + AB^* + BA^*$

戻り値

なし

計算量

$O(\verb|A.extent(0)| \times \verb|A.extent(1)| \times \verb|B.extent(0)|)$

備考

  • (1), (2)は出力行列Cを上書きする。従来の加算更新 ($C \leftarrow C + AB^ + BA^$) を行いたい場合は、updating版 (3), (4)に元の行列をEとして渡す。
  • エルミート行列Cの対角成分については、real-if-neededにより実部のみが使用される。対角成分が非ゼロの虚部を持っていても、その虚部は無視される。

例

[注意] 処理系にあるコンパイラで確認していないため、間違っているかもしれません。

#include <array>
#include <iostream>
#include <linalg>
#include <mdspan>
#include <vector>
#include <complex>

template <class Matrix>
void print_mat(const Matrix& A) {
  for(int i = 0; i < A.extent(0); ++i) {
    for(int j = 0; j < i; ++j) {
      std::cout << A[j, i] << ' ';
    }
    for(int j = i; j < A.extent(1) - 1; ++j) {
      std::cout << A[i, j] << ' ';
    }
    std::cout << A[i, A.extent(1) - 1] << '\n';
  }
}

template <class Matrix>
void init_mat(Matrix& A, double geta = 0.0) {
  for(int i = 0; i < A.extent(0); ++i) {
    for(int j = 0; j < A.extent(1); ++j) {
      A[i,j] = std::complex<double>(i + geta, j + geta);
    }
  }
}

template <class Matrix>
void init_herm_mat(Matrix& A) {
  for(int i = 0; i < A.extent(0); ++i) {
    A[i, i] = std::complex<double>(i, 0);
    for(int j = i + 1; j < A.extent(1); ++j) {
      A[i, j] = std::complex<double>(i, j);
    }
  }
}

int main()
{
  constexpr std::size_t N = 2;

  using Scalar = std::complex<double>;
  using Vector = std::vector<Scalar>;
  using Matrix = std::mdspan<Scalar, std::extents<std::size_t, N, N>>;
  using HermitianMatrix = std::mdspan<
      Scalar,
      std::extents<std::size_t, N, N>,
      std::linalg::layout_blas_packed<
        std::linalg::upper_triangle_t,
        std::linalg::row_major_t>
    >;

  Vector A_vec(N * N);
  Vector B_vec(N * N);
  Vector E_vec(N * N);
  Vector C_vec(N * N);

  Matrix A(A_vec.data());
  Matrix B(B_vec.data());
  HermitianMatrix E(E_vec.data());
  HermitianMatrix C(C_vec.data());

  init_mat(A);
  init_mat(B, 1);
  init_herm_mat(E);

  // (1) overwriting: C = A B^* + B A^*
  std::cout << "(1)\n";
  std::linalg::hermitian_matrix_rank_2k_update(
    A,
    B,
    C,
    std::linalg::upper_triangle);
  print_mat(C);

  // (3) updating: C = E + A B^* + B A^*
  std::cout << "(3)\n";
  std::linalg::hermitian_matrix_rank_2k_update(
    A,
    B,
    E,
    C,
    std::linalg::upper_triangle);
  print_mat(C);

  return 0;
}

出力

(1)
(4,0) (6,4)
(6,4) (12,0)
(3)
(4,0) (6,5)
(6,5) (13,0)

バージョン

言語

  • C++26

処理系

関連項目

参照