Question

我想使用std :: vector定义n维数据结构。我对operator（）的定义有问题。让我们看一下2D结构样本

class my_data {

public:
  my_data (size_t N, size_t M) : tab(N*M), _N(N), _M(M) {}

  const double& operator() (size_t i, size_t j) const {
     return tab.at(i * M + j);
  }

  double& operator() (size_t i, size_t j) {
     return tab.at(i * M + j);
  }

private:
   std::vector<double> tab;
   size_t _N;
   size_t _M;

};

我想使用模板为任何维度定义这样的结构，但我不知道是否可能。基本上我想拥有的就是那样的东西

my_data<4> vec (1,2,3,4);

定义尺寸为1,2,3,4的4D'张量'（所有元素的数量为1 * 2 * 3 * 4）。您现在可以通过键入

来访问和更改值

vec (0,0,0,0) = 2.0;

Answer 1

以下解决方案使用了一些C ++ 14和C ++ 1z功能，但它们可以轻松移植到C ++ 11：

#include <vector>
#include <utility>
#include <array>
#include <cstddef>

namespace detail
{
    template <typename T, typename S>
    class my_data;

    template <typename T, std::size_t... Is>
    class my_data<T, std::index_sequence<Is...>>
    {
    public:
        my_data(decltype(Is)... size)
            : t((size * ...)), s{{ size... }}
        {}

        T& operator()(decltype(Is)... i)
        {            
            return t.at(index({{ i... }}));
        }

        const T& operator()(decltype(Is)... i) const
        {            
            return t.at(index({{ i... }}));
        }

    private:
        std::size_t index(const std::array<std::size_t, sizeof...(Is)>& a) const
        {
            std::size_t ind = a[0];
            for (std::size_t i = 1; i < a.size(); ++i)
            {
                ind = ind * s[i] + a[i];
            }
            return ind;
        }

        std::vector<T> t;
        const std::array<std::size_t, sizeof...(Is)> s;
    };
}

template <typename T, std::size_t N>
using my_data = detail::my_data<T, std::make_index_sequence<N>>;

测试：

int main()
{        
    my_data<double, 4> vec(1,2,3,4);
    vec(0,0,0,0) = 2.0;
}

DEMO

DEMO (C++11)

基于std :: vector的N维张量

1 个答案: