c++ - 通过提供 rowIndices 列表从矩阵中获取行

Question

我是 xtensor 的初学者，我目前正在寻找从数组列表中获取行的方法。

我有以下矩阵。

auto matrix = {{  0.,   1.,   0.,   1.,   1.,   1.,   1.},
               {  1.,   2.,   0.,   1.,   3.,   1.,   1.},
               {  2.,   3.,   0.,   2.,   7.,   0.,   0.},
               {  3.,   4.,   0.,   1.,  11.,   0.,   1.},
               {  4.,   0.,   1.,   1.,   0.,   0.,   0.}}

我想从这个矩阵中选择以下行。

xt::xarray<int> rowIndices = { 1, 2, 3, 4 }

现在我想使用这个 rowIndices 数组来获取包含所有行的子矩阵。我怎样才能做到这一点？

我尝试了以下方法。

xt::view(matrix, rowIndices, xt::all())

但这行不通。

Answer 1

您需要使用 xt::keep(...) 按索引选择行。

完整示例:

#include <xtensor/xtensor.hpp>
#include <xtensor/xview.hpp>
#include <xtensor/xio.hpp>

int main()
{
  xt::xtensor<double,2> a =
    {{  0.,   1.,   0.,   1.,   1.,   1.,   1.},
     {  1.,   2.,   0.,   1.,   3.,   1.,   1.},
     {  2.,   3.,   0.,   2.,   7.,   0.,   0.},
     {  3.,   4.,   0.,   1.,  11.,   0.,   1.},
     {  4.,   0.,   1.,   1.,   0.,   0.,   0.}};

  xt::xtensor<size_t,1> rowIndices = { 1, 2, 3, 4 };

  auto v = xt::view(a, xt::keep(rowIndices), xt::all());

  std::cout << v << std::endl;

  return 0;
}

打印:

{{  1.,   2.,   0.,   1.,   3.,   1.,   1.},
 {  2.,   3.,   0.,   2.,   7.,   0.,   0.},
 {  3.,   4.,   0.,   1.,  11.,   0.,   1.},
 {  4.,   0.,   1.,   1.,   0.,   0.,   0.}}

请注意，根据 documentation ，鉴于您还可以使用 xt::range(...)、xt::all()、xt::newaxis()、xt::keep(...) 和 xt::drop(...)。