Question

我想有效地将参数包中的参数与std :: array的元素相乘：

int index(auto... Is, std::array<int,sizeof...(Is)> strides)
{
  // pseudo-code
  // int idx = 0;
  // for(int i = 0; i < sizeof...(Is); ++i)
  //   idx += Is[i] * strides[i];
  // return idx; 
}

我无法完全围绕这一个。我开始沿着索引序列前进，但我可以弄清楚如何合并索引。

我使用的是c ++ 17，因此如果简化代码，折叠表达式就是公平的游戏。

感谢您的任何指示。

编辑：澄清了伪代码。唯一的伪部分是表达式Is[i]，它引用第i个参数包参数。

T.C.下面的答案是完美的，这是我的最终代码，它是一个成员函数：

unsigned int index(auto... indexes)
{
    unsigned int idx = 0, i = 0;
    (..., (idx += indexes * m_strides[i++]));
    return idx;
}

在撰写本文时，代码使用带有-fconcepts标志的gcc 6.3.0进行编译，该标志引入了Concept TS。

使用auto... indexes是template<typename Args> f(Args... indexes)的简写。我尝试使用unsigned int概念作为参数，但我无法让它工作。

（...，）fold是关键元素，并扩展为类似的东西（如果你实际上可以[]进入参数包中）：

idx += indexes[0] * m_strides[i++], idx += indexes[1] * m_strides[i++], etc.

这就是我所缺少的洞察力。

Answer 1

我无法让auto...工作，因此我更改了index的签名。

您需要一个辅助函数（此处为index_helper）才能使用index_sequence，因为它依赖模板参数推导来填充索引。

#include <array>
#include <cstdio>

template <typename... T, size_t... i>
//                       ^~~~~~~~~~~
//                        use deduction to make {i...} = {0, 1, 2, ..., n}
static int index_helper(const std::array<int, sizeof...(T)>& strides,
                        std::index_sequence<i...>,
                        T... Is) 
{
    return (0 + ... + (strides[i] * Is));
}

template <typename... T>
int index(const std::array<int, sizeof...(T)>& strides, T... Is) {
    return index_helper(strides, std::make_index_sequence<sizeof...(T)>(), Is...);
//                               ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//                                generates {0, 1, 2, ..., n}
}

int main() {
    printf("%d\n", index({1, 100, 100000, 1000}, 2, 3, 5, 7));
    // 507302
}

Answer 2

如果您可以将参数包打包成一个复制/移动便宜的单一类型，您可以将它变成一个数组：

T arr[] = { static_cast<T>(Is)... }; // for some T, possibly common_type_t<decltype(Is)...>

然后你可以把你的伪代码变成真正的代码。

如果这不可行，可以使用逗号折叠：

int idx = 0, i = 0;
(..., (idx += Is * strides[i++]));
return idx;

c ++ 17有效地将参数包参数与std :: array elements

2 个答案: