0

我有以下代码float，例如：

std::vector<float> v = {0.f, 1.f, 2.f};
for(size_t i = 0; i < v.size(); ++i)
    if(v[i] != 0) // An optimization for `v[i] != 0`.
        v[i] = v[i] * v[i] * v[i]; // Time-consuming computation.

现在我想替换float为std::valarray<float>：

using vfloat = std::valarray<float>;
std::vector<vfloat> v = {vfloat{0.f, 0.f}, vfloat{1.f, 0.f}, vfloat{2.f, 0.f}};
for(size_t i = 0; i < v.size(); ++i)
    // if(v[i] != 0) // I want this optimization! ◀◀◀◄◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀◀
        v[i] = v[i] * v[i] * v[i]; // Time-consuming computation.

虽然新代码在逻辑上是正确的，但对于v[i][j] == 0.

那么如何在没有冗余计算的情况下使用std::valarrayinif语句呢？

4

2 回答 2

1

如果你想跳过v[i]使用嵌套循环的元素

using vfloat = std::valarray<float>;
std::vector<vfloat> v = {vfloat{0.f, 0.f}, vfloat{1.f, 0.f}, vfloat{2.f, 0.f}};
for(size_t i = 0; i < v.size(); ++i)
    for(size_t j = 0; j < v[i].size(); ++j)
        if (v[i][j] != 0)
            v[i][j] = v[i][j] * v[i][j] * v[i][j]; // Time-consuming computation.

或apply：

using vfloat = std::valarray<float>;
std::vector<vfloat> v = {vfloat{0.f, 0.f}, vfloat{1.f, 0.f}, vfloat{2.f, 0.f}};
for(size_t i = 0; i < v.size(); ++i)
    v[i] = v[i].apply([](float n) -> float {
        if (n == 0) return 0;
        return n * n * n; // Time-consuming computation.
    });

如果你想跳过元素v 

using vfloat = std::valarray<float>;
std::vector<vfloat> v = {vfloat{0.f, 0.f}, vfloat{1.f, 0.f}, vfloat{2.f, 0.f}};
for(size_t i = 0; i < v.size(); ++i)
    if ((v[i] != 0).max() != 0)
        v[i] = v[i] * v[i] * v[i]; // Time-consuming computation.
于 2020-05-31T15:53:49.330 回答
0

有很多选择：

#include <valarray>
#include <algorithm>
// 2 loops in the worst case 
bool is_not_zero_0(std::valarray<float> const & v)
{
    return v.max() != 0.0 || v.min() != 0.0;
}

// one loop
bool is_not_zero_1(std::valarray<float> const & v)
{
    return std::find_if(v.begin(), v.end(v), [](float item){return item != 0.0f;}) != v.end();
}
// 2 loops + allocation
bool is_not_zero_2(std::valarray<float> const & v)
{
    return (v != 0.0).max();
}
于 2020-05-31T15:42:47.420 回答