天真的解决方案

Question

请参阅下面的代码段（矩阵乘法的实现）。
是否可以使用nested pack expansion来简化它们，使其具有类似{{((a[r][k] * b[k][c]) + ...)...}...}的内容？

#include <array>
#include <utility>

template<typename T, size_t R, size_t C>
using Matrix = std::array<std::array<T, C>, R>;

template<typename A, typename B>
using mul_el_t = decltype(std::declval<A>()[0][0] * std::declval<B>()[0][0]);

帮助计算单个元素。

template<size_t R1, size_t C2, size_t... C1_R2, typename A, typename B>
auto _mat_mul_element(const A &a, const B &b, std::index_sequence<C1_R2...>)
{
    return ((a[R1][C1_R2] * b[C1_R2][C2]) + ...);
}

帮助计算特定行。

template<size_t R1, size_t... C2, typename C1_R2, typename A, typename B>
auto _mat_mul_row(const A &a, const B &b, std::index_sequence<C2...>, C1_R2 c1_r2)
-> std::array<mul_el_t<A, B>, sizeof...(C2)>
{
    return {_mat_mul_element<R1, C2>(a, b, c1_r2)...};
}

这使用参数包计算整个矩阵。

template<size_t... R1, typename C2, typename C1_R2, typename A, typename B>
auto _mat_mul(const A &a, const B &b, std::index_sequence<R1...>, C2 c2, C1_R2 c1_r2)
-> Matrix<mul_el_t<A, B>, sizeof...(R1), C2::size()>
{
    return {_mat_mul_row<R1>(a, b, c2, c1_r2)...};
}

实际界面。

template<typename T, size_t R1, size_t C1_R2, size_t C2>
Matrix<T, R1, C2> operator*(const Matrix<T, R1, C1_R2> &a, const Matrix<T, C1_R2, C2> &b)
{
    return _mat_mul(
            a, b,
            std::make_index_sequence<R1>{},
            std::make_index_sequence<C2>{},
            std::make_index_sequence<C1_R2>{}
    );
};

更新（看起来我不清楚我遇到的实际问题）

当我尝试将_mat_mul替换为：

时

template<size_t... R1, size_t...  C2, size_t...  C1_R2, typename A, typename B>
auto _mat_mul(const A &a, const B &b,
                std::index_sequence<R1...>,
                std::index_sequence<C2...>,
                std::index_sequence<C1_R2...>)
-> Matrix<mul_el_t<A, B>, sizeof...(R1), sizeof...(C2)>
{
    return {{((a[R1][C1_R2] * b[C1_R2][C2]) + ...)...}...};
}

使用Apple LLVM version 9.1.0 (clang-902.0.39.1)编译失败：

[ 50%] Building CXX object CMakeFiles/main.cpp.o
main.cpp:38:51: error: pack expansion does not contain any unexpanded parameter packs
    return {{((a[R1][C1_R2] * b[C1_R2][C2]) + ...)...}...};
             ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^

我认为失败是预期的，因为编译器不知道在每个扩展块中扩展哪个包（R1，C2或C1_R2）。在这种情况下如何提示编译器（注意，我可以使用任何编译器）？

Answer 1

根据文档，嵌套包扩展可以看作是一个迭代过程，从最里面的包扩展[3点]开始。每个包扩展都会扩展子包含扩展中包含的所有参数包。

因此，{{((a[R1][C1_R2] * b[C1_R2][C2]) + ...)...}...}在第一步后成为{{(a[0][0] * b[0][0] + a[1][1] * b[1][1])...}...}（R1/C2/C1_R2为index_sequence<2>）。因此，接下来的两包扩展只是没有任何扩展。

天真的解决方案

可以将每个参数包同时移动到所需的子表达式，并将其携带的实际值保留在所需的位置。可以使用FP的let ... in表达式的模拟：

auto let = [](auto a, auto f) { return f(a); };

因此原始表达变为：

{let(R1, [&](auto r1) {
    return std::array<T, sizeof...(C2)>{let(C2, [&](auto c2) {
        return ((a[r1][C1_R2] * b[C1_R2][c2]) + ...);
    })...};
})...};

这可能已经足够好了，但可能需要一些时间来破译那里发生的事情。还需要在范围内引入这些参数包。

更好的解决方案

可以尝试通过抽象参数包引入/扩展来提高可读性。使用以下实用功能。

template<typename H, typename F, typename T, T... I>
decltype(auto) repack(std::integer_sequence<T, I...>, H h, F f)
{
    return h(f(std::integral_constant<T, I>{})...);
}

此函数采用带有一些包的值（一个可以对除std::integer_sequence之外的其他内容进行重载），函数f应用于包的每个元素，并且函数{{1} }用于将最终包转换为某个值。

因此，完全乘法例程变为

其中template<typename T, size_t R1, size_t C1_R2, size_t C2> Matrix<T, R1, C2> operator*(const Matrix<T, R1, C1_R2> &a, const Matrix<T, C1_R2, C2> &b) { std::make_index_sequence<R1> r1{}; std::make_index_sequence<C2> c2{}; std::make_index_sequence<C1_R2> c1_r2{}; return repack(r1, ctor<Matrix<T, R1, C2>>(), [&](auto r1) { return repack(c2, ctor<std::array<T, C2>>(), [&](auto c2) { return repack(c1_r2, sum, [&](auto c1_r2) { return a[r1][c1_r2] * b[c1_r2][c2]; }); }); }); }是

ctor

和template<typename H> auto ctor() { return [](auto... xs) { return H{xs...}; }; }。

疯狂解决方案

有人可能在3个嵌套sum = [](auto... xs) { return (xs +...); };的表达式中有斑点模式，因此乘法例程可能变为：

repack

使用实用程序：

template<typename T, size_t R1, size_t C1_R2, size_t C2>
Matrix<T, R1, C2> operator*(const Matrix<T, R1, C1_R2> &a, const Matrix<T, C1_R2, C2> &b)
{
    auto item = [&](auto r1, auto c2, auto c1_r2) { return a[r1][c1_r2] * b[c1_r2][c2]; };
    auto curried_repack = curry(POLY(repack));
    auto m = curried_repack(std::make_index_sequence<R1>{}, ctor<Matrix<T, R1, C2>>());
    auto r = curried_repack(std::make_index_sequence<C2>{}, ctor<std::array<T, C2>>());
    auto e = curried_repack(std::make_index_sequence<C1_R2>{}, sum);

    auto op = [](auto w, auto f) {
        return compose(w, curry(f));
    };
    return foldr(op, m, r, e, item)();
}

宏将模板函数转换为值

template<typename F>
auto curry(F f)
{
    return [=](auto... a) {
        return [=](auto... b) { return f(a..., b...); };
    };
};

template<typename F, typename G>
auto compose(F f, G g)
{
    return [=](auto... xs) {
        return f(g(xs...));
    };
};

#define POLY(f) ([](auto... a){ return f(a...); })作为家庭作业。

编译说明

所有解决方案在等同意义上都是相同的二进制文件。

嵌套参数包扩展

1 个答案:

天真的解决方案

更好的解决方案

疯狂解决方案

编译说明