テンソルの行、列、および面をどのように反復するのですか？

Question

私はカスタム演算子の出力テンソルを反復しようとしています。これまでのところ、私はテンソルを平坦化し、それを反復するアプローチしか見ません。

もっと良い方法があるので、私は行、列、および面を反復処理できますか？

auto output = output_tensor->flat<float>(); 

// I would like to iterate through the dimensions here 
const int N = output.size(); 
for (int i = 0; i < N; i++) { 
    output(i) = 0; 

Answer 1

固有テンソルライブラリが文書化されているが、このページの下部に食べた：あなたの質問から

https://bitbucket.org/eigen/eigen/src/f6382682565c946d46612fe0e36e486bba1371ce/unsupported/Eigen/CXX11/src/Tensor/?at=default

を、私はあなたがチップとスライス操作を見てみたいと思います。