如何迭代張量的行，列和平面？

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

從你的問題，我想你會想看看在芯片和切片操作。