c++ - 如何在 xt::filter 中使用相等？

Answer 1

您可以使用xt::equal(a, b)而不是a == b. IE

xt::filter(a, xt::equal(a, 2)) = 10;

做你想做的事。

Answer 2

xtensor 提供逻辑运算符，如&&and ||。将这些与允许的比较相结合会产生与 '==' 或 '!=' 所期望的相同的输出。

特别是a >= 2 && a <= 2<=>a == 2和a > 2 || a < 2<=>a != 2

所以我的最终程序是

#include <iostream>

#include "xtensor/xadapt.hpp"
#include "xtensor/xarray.hpp"
#include "xtensor/xindex_view.hpp"
#include "xtensor/xio.hpp"
#include "xtensor/xmasked_view.hpp"
#include "xtensor/xview.hpp"

using namespace std;

int main() {
    xt::xarray<float> a = {{1, 2, 3}, {4, 2, 6}, {9, 0, 2}};
    cout << a << endl;
    xt::filter(a, a >= 2 && a <= 2) = 10;
    cout << a << endl;
}

它的输出是

{{ 1.,  2.,  3.},
 { 4.,  2.,  6.},
 { 9.,  0.,  2.}}
{{  1.,  10.,   3.},
 {  4.,  10.,   6.},
 {  9.,   0.,  10.}}