Eigen::Tensor，如何從張量訪問矩陣

Question

我有以下特征張量：

Eigen::Tensor<float, 3> m(3,10,10);

我想訪問第一個矩陣。 在 numpy 我會這樣做

m(0,:,:)

我將如何在 Eigen 中做到這一點

Answer 1

您可以使用.slice(...)或.chip(...)訪問張量的.slice(...) 。 這樣做可以訪問第一個矩陣，相當於 numpy m(0,:,:) ：

Eigen::Tensor<double,3> m(3,10,10);          //Initialize
m.setRandom();                               //Set random values 
std::array<long,3> offset = {0,0,0};         //Starting point
std::array<long,3> extent = {1,10,10};       //Finish point
std::array<long,2> shape2 = {10,10};         //Shape of desired rank-2 tensor (matrix)
std::cout <<  m.slice(offset, extent).reshape(shape2) << std::endl;  //Extract slice and reshape it into a 10x10 matrix.

如果您想要“第二個”矩陣，則使用offset={1,0,0}代替，依此類推。

您可以 在此處找到 最新的文檔。