0

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

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 类以添加功能?

4

0 回答 0