如何声明特征矩阵，然后通过嵌套循环初始化它

How to declare Eigen matrix and then initialize it through nested loop

本文关键字：然后 嵌套循环 初始化 何声明 声明 特征      更新时间：2023-10-16

我想做标题中写的事情，我尝试了以下代码：

typedef std::vector<std::vector<std::vector<double>>> Tensor;
// should return a matrix of shape (batch_size, 1840)
Eigen::MatrixXd flatten(Tensor x)
{
int channels = x.size(); // always 10
int batch = x[0].size(); // varies
int columns = x[0][0].size(); // always 184
Eigen::Matrix<double, Eigen::Dynamic, 1840> matrix;
for (unsigned int b = 0; b < batch; b++)
{
for (unsigned int i = 0; i < columns; i++)
{
for (unsigned int c = 0; c < channels; c++)
{
matrix << x[c][b][i]; // I also tried this: matrix(b, i+c) = x[c][b][i];
}
}
}
return matrix;
}

但代码要么中止abort() has been called消息，要么给我一个Access violation writing location 0x0000000000000000

完成我想做的事情的正确方法是什么？

你从来没有告诉矩阵应该是什么大小。在写入矩阵之前，你必须调整矩阵的大小。

Eigen::Matrix<double, Eigen::Dynamic, 1840> matrix;
matrix.resize(batch, 1840);
for (unsigned int b = 0; b < batch; b++)
...

代码的第二个问题是operator<<的行为与std容器的push_back不同。它一次初始化整个(大小合适的(矩阵，而不是使用NxMxL调用。

与代码无关的一个性能问题是，Tensor x是按值传递的，从而导致复制。代之以传递(const(引用：

Eigen::MatrixXd flatten(const Tensor& x)
{ ...
相关文章：