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

Question

以下代码，

#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) = 10;
    cout << a << endl;
}

无法编译以下内容:error: no match for ‘operator==’ (operand types are ‘xt::xarray<float>’ ... and ‘int’)

但是，其他比较运算符按预期工作(>、<、>=、<=)。我不确定operator==是否是故意不实现的，但在它实现之前(如果曾经实现过)，是否有解决方法，它是什么？

Answer 1

您可以使用xt::equal(a, b)代替a == b。即

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

做你想做的事。