Opencv张量乘积

Opencv tensor product

本文关键字：张量乘 Opencv      更新时间：2023-10-16

我有两个2x2矩阵，我想用opencv得到它们的张量积，例如：

{
{0, 1},
{1, 0}
}
x
{
{1, 0},
{0, 1}
}
=
{
{0, 0, 1, 0},
{0, 0, 0, 1},
{1, 0, 0, 0},
{0, 1, 0, 0}
}

所以我有这样的东西：

cv::Mat mat1 = (cv::Mat_<double>(2,2) << 1, 0, 0, 1);
cv::Mat mat2 = (cv::Mat_<double>(2,2) << 0, 1, 1, 0);

有没有在opencv中内置一些张量积函数来获得我需要的结果？

OpenCV不提供张量上的内置操作。您需要依靠其他库来实现这一点。

然而，我不是张量的专家，但对我来说，这看起来像是Kronecker产品（在OpenCV中仍然没有）。如果是这样的话，你可以在这里找到一个实现，我也在下面的代码中报告了它，它产生了你需要的结果：

[0, 1;
 1, 0]
 x
[1, 0;
 0, 1]
 =
[0, 0, 1, 0;
 0, 0, 0, 1;
 1, 0, 0, 0;
 0, 1, 0, 0]

代码：

#include <opencv2/opencv.hpp>
using namespace cv;
using namespace std;
Mat kron(const Mat& A, const Mat& B)
{
    CV_Assert(A.channels() == 1 && B.channels() == 1);
    Mat1d Ad, Bd;
    A.convertTo(Ad, CV_64F);
    B.convertTo(Bd, CV_64F);
    Mat1d Kd(Ad.rows * Bd.rows, Ad.cols * Bd.cols, 0.0);
    for (int ra = 0; ra < Ad.rows; ++ra)
    {
        for (int ca = 0; ca < Ad.cols; ++ca)
        {
            Kd(Range(ra*Bd.rows, (ra + 1)*Bd.rows), Range(ca*Bd.cols, (ca + 1)*Bd.cols)) = Bd.mul(Ad(ra, ca));
        }
    }
    Mat K;
    Kd.convertTo(K, A.type());
    return K;
}
int main(int argc, char **argv)
{
    Mat mat1 = (Mat_<double>(2, 2) << 0, 1, 1, 0);
    Mat mat2 = (Mat_<double>(2, 2) << 1, 0, 0, 1);
    Mat res = kron(mat1, mat2);
    cout << mat1 << endl << endl;
    cout << " x " << endl << endl;
    cout << mat2 << endl << endl;
    cout << " = " << endl << endl;
    cout << res << endl;
    return 0;
}
相关文章：