Question

基于@Piotr Skotnicki回答here我定义了以下数据类型：

#ifndef TENSOR_HPP
#define TENSOR_HPP

#include <vector>
#include <array>
#include <boost/numeric/odeint.hpp>

namespace nex {

template <size_t...> struct seq {};
template <size_t N, size_t... Is> struct gen_seq : gen_seq<N-1, N-1, Is...> {};
template <size_t... Is> struct gen_seq<0, Is...> { using type = seq<Is...>; };

template<typename T, typename S>
class tensor;

template<typename T, size_t... Is>
class tensor<T, seq<Is...>>
{

    typedef std::vector<T> vector;

public:

    // constructor
    tensor ();

    tensor (decltype(Is)...);

    // iterators
    using iterator       =  typename vector::iterator;
    using const_iterator =  typename vector::const_iterator;

    iterator begin()
    { return m_v.begin(); }

    const_iterator begin() const
    { return m_v.begin(); }

    iterator end() 
    { return m_v.end(); }

    const_iterator end() const
    { return m_v.end(); }

    // resizing
    size_t size() const; 

    void resize( const size_t  );

    // operators

    const T& operator() (decltype(Is)...) const;

    T& operator() (decltype(Is)...); 

private:

    // helper functions
    template <size_t I>
    auto index(const std::array<size_t, sizeof...(Is)>& a) const
    -> typename std::enable_if<I == 0, size_t>::type 
    {
        return a[I];
    }

    template <size_t I>
    auto index(const std::array<size_t, sizeof...(Is)>& a) const
    -> typename std::enable_if<I != 0, size_t>::type 
    {
        return index<I-1>(a) * s[I] + a[I];
    }

    size_t mult(size_t N) 
    {
        return N;
    }

    template <typename... S>
    size_t mult(size_t N, S... Ns) 
    {
        return N * mult(Ns...);
    }

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

    }; // class tensor

 template<typename T, size_t... Is    
 tensor<T, seq<Is...>> :: tensor ()
   : m_v(), s{ { 0 } }
{}

template<typename T, size_t... Is>
tensor<T, seq<Is...>> :: tensor (decltype(Is)... size)
   : m_v( mult( size... ) ), s{ { size... } }
{}

template<typename T, size_t... Is>
size_t tensor<T, seq<Is...>> :: size () const 
{ return m_v.size(); }

template<typename T, size_t... Is>
void tensor<T, seq<Is...>> :: resize (const size_t n)
{ m_v.resize( n ); }

template<typename T, size_t... Is>
const T& tensor<T, seq<Is...>> :: operator() (decltype(Is)... n) const
{ return m_v.at(index<sizeof...(Is)-1>( { { n... } } ) ); }

template<typename T, size_t... Is>    
T& tensor<T, seq<Is...>> :: operator() (decltype(Is)... n) 
{ return m_v.at(index<sizeof...(Is)-1>( { { n... } } ) ); }

} // namespace nex

namespace boost { namespace numeric { namespace odeint {

template<typename T, size_t... Is>
struct is_resizeable< nex::tensor<T, nex::seq<Is...>> >
{
    typedef boost::true_type type;
    static const bool value = type::value;
};
} // namespace odeint
} // namespace numeric
} // namespace boost

template <typename T, std::size_t N>
using tensor = nex::tensor<T, typename nex::gen_seq<N>::type>;

#endif // TENSOR_HPP

然后我想在boost :: odeint中使用这种数据类型来集成非常基本的问题

#include "tensor.hpp"

typedef tensor<double, 1> state_type;

void lorenz( const state_type &x , state_type &dxdt , const double t )
{

    dxdt(0) = x(0);
    dxdt(1) = x(1);
    dxdt(2) = x(2);
}

void write_lorenz( const state_type &x , const double t )
{
    std::cout << t << '\t' << x(0) << '\t' << x(1) << '\t' << x(2) << std::endl;
}


using namespace boost::numeric::odeint;

int main()
{
    state_type x(3);
    x(0) = 1.0 ; x(1) = 10.0 ; x(2) = 10.0;


    integrate_const( runge_kutta4< state_type >() , lorenz , x , 0.0 , 1.0 , 0.1, write_lorenz );
}

它工作正常，即使对于不同的方程组（如洛伦兹吸引子）。当我想使用相同的张量容量但定义不同时，不幸的是每次都出错了

#include "tensor.hpp"

typedef tensor<double, 2> state_type;

void lorenz( const state_type &x , state_type &dxdt , const double t )
{

    dxdt(0,0) = x(0,0);
    dxdt(1,0) = x(1,0);
    dxdt(2,0) = x(2,0);
}

void write_lorenz( const state_type &x , const double t )
{
    std::cout << t << '\t' << x(0,0) << '\t' << x(1,0) << '\t' << x(2,0) << std::endl;
}


using namespace boost::numeric::odeint;

int main()
{
    state_type x(3,1);
    x(0,0) = 1.0 ; x(1,0) = 10.0 ; x(2,0) = 10.0;


    integrate_const( runge_kutta4< state_type >() , lorenz , x , 0.0 , 1.0 , 0.1, write_lorenz );
}

您可以自己运行代码并检查x（1,0）x（2,0）是否完全没有变化。我虽然对operator（）的适当定义足以让odeint正常工作。我对boost :: odeint的了解很少，所以这可能只是一个基本的bug。我指望你的帮助。

更新

当我使用std :: array将张量定义为固定大小的数组时，它可以正常工作。但后来我不需要添加boost :: is_resizeable。我认为调整大小需要以不同的方式定义（我不确切地知道）。

Answer 1

事实上，问题在于调整大小。目前，调整张量的大小只会改变基础向量<script type="text/javascript"> var now = new Date(); if (now.getHours() > 11 && now.getHours() < 14) { //popup code } </script>的大小。虽然这可以确保正确的内存分配，但这还不够。您还需要正确设置尺寸，即成员m_v。您也可以这样思考：只给出一个整数就不可能调整多维张量的大小，您需要设置每个维度的大小。唯一的例外是1D张量，对于你来说，事情很好，对于任何更高的维度，这都必须失败。

以下tensor.hpp解决了这个问题，似乎在我的快速测试中运行良好：

具有自定义张量数据结构的boost :: odeint输出错误

更新

1 个答案: