我正在为张量编写一个类,我的工作需要它。我想出了如何使用递归创建动态多维矢量的方法,但是我不知道如何将其拉伸到所需的尺寸(如vector<double> vec; vec.resize(6);
)。但是我想获得动态矢量并创建了模板类。
有人可以帮我吗?这是我的代码:
template<typename T,size_t N>
struct tensor_traits
{
using type = vector< typename tensor_traits<T,N-1>::type>;
};
template<typename T>
struct tensor_traits<T,0>
{
using type = T;
};
template<class T,size_t N>
class tenzor
{
protected:
typename tensor_traits<T, N>::type data;
vector<size_t> dimensions;
public:
tenzor()
{
dimensions.resize(N);
}
tenzor(vector<size_t> dim)
{
if(dim.size() == N)
{
dimensions = dim;
data.resize(dim);
}
else
throw::length_error("wrong format!!");
}
void resize(vector<size_t> siz)
{
//some code`enter code here`
}
}