Question

我最初有一个3x3的矩阵，我将其展平（所以现在我有一个1D数组，从初始的3x3矩阵开始一行接一行），我必须将其乘以3x1向量。我该怎么办？我不知道该怎么做迭代。我正在用C ++做到这一点。

Answer 1

这是使用扁平矩阵的非常简单的解决方案，并且没有任何边界检查以使其简单：

#include <iostream>
#include <vector>

struct matrix2d
{
    std::vector<float> v_;
    size_t x_, y_;
};

std::vector<float> mult3x3(const matrix2d &m, const std::vector<float> &v)
{
    std::vector<float> result;
    for (size_t i=0; i< m.y_; i++)
    {
        float r = 0.0;
        for (size_t j=0; j< m.x_; j++)
        r += m.v_[i*m.x_+j]*v[j];
        result.push_back(r);
    }
    return result;
}

int main()
{
    matrix2d m {.v_ = {1,2,3, 1,2,3, 1,2,3}, .x_=3, .y_=3};
    std::vector<float> v { 1,2,1};

    std::vector<float> result = mult3x3(m, v);
    for (size_t i=0; i< result.size(); i++)
        std::cout << result[i] << ", " << std::endl;
    return 0;
}

See it here

将矩阵乘以向量

1 个答案: