我创建了一个函数来显示特征上张量的形状。
template<typename Scalar_, int rank>
void shape(const Eigen::Tensor<Scalar_, rank>& x)
{
cout << "( ";
for (int i(0); i<x.NumDimensions; i++){
cout << x.dimensions()[i];
cout << ",";
}
cout << ")";
}
要使用它,我必须这样做:
Tensor <double,2> t (2,2)
shape(t)
我想将它直接集成到张量类中,这样我们就可以这样称呼它:
Tensor <double,2> t (2,2)
t.shape
如何扩展 Eigen 的 Tensor 类以添加功能?