Добавьте SparseMatrix для Eigen в тензор Eigen
Как я могу добавить Eigen’s SparseMatrix к Eigen’s Tensor?
Следующий код (который не компилируется) объясняет, что я пытаюсь сделать.
#include <iostream>
#include <Eigen/Sparse>
#include <unsupported/Eigen/CXX11/Tensor>
using Eigen::Tensor;
using Eigen::SparseMatrix;
int main()
{
Tensor<double, 2> tensor(10, 10);
for(int i=0; i < 10; i++) {
for(int j=0; j < 10; j++) {
tensor(i, j) = i * 10 + j;
}
}
SparseMatrix<double> sparse(10, 10);
auto tensor2 = tensor;
tensor2 += sparse;
std::cout << tensor2 << std::endl;
}
Решение
Очевидно, это не реализовано. Вы должны перегрузить operator+= для этих двух типов самостоятельно. Увидеть этот стол для правильной подписи. Смотрите также »Итерации по ненулевым коэффициентам« в собственных документах о том, как эффективно перебирать разреженную матрицу.
#include <iostream>
#include <Eigen/Sparse>
#include <unsupported/Eigen/CXX11/Tensor>
using Eigen::Tensor;
using Eigen::SparseMatrix;
template < typename T >
Tensor<T,2>& operator+=(Tensor<T,2>& lhs, SparseMatrix<T> const& rhs)
{
for (int k = 0; k < rhs.outerSize(); ++k)
for (typename SparseMatrix<T>::InnerIterator it(rhs,k); it; ++it)
lhs(it.row(), it.col()) = it.value();
return lhs;
}
int main()
{
Tensor<double, 2> tensor(10, 10);
for(int i=0; i < 10; i++) {
for(int j=0; j < 10; j++) {
tensor(i, j) = i * 10 + j;
}
}
// We want a sparse matrix that is not only zeros
Eigen::MatrixXd m = Eigen::MatrixXd::Zero(10,10);
m(0,0) = 1;
SparseMatrix<double> sparse(10, 10);
sparse = m.sparseView();
auto tensor2 = tensor;
tensor2 += sparse;
std::cout << tensor2 << std::endl;
}
Другие решения
Вы имеете в виду в += метод:
lhs(it.row(), it.col()) += it.value();
и не
lhs(it.row(), it.col()) = it.value();
?