如何扩展 Eigen 的 Tensor 类?

问题描述

我创建了一个函数显示特征上张量的形状。

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.shape

如何扩展 Eigen 的 Tensor 类以添加函数

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)