Làm thế nào để lặp qua các hàng, cột và mặt phẳng của một tensor?

Question

Tôi đang cố gắng lặp qua đầu ra tensor trong một toán tử tùy chỉnh. Cho đến nay tôi chỉ thấy cách tiếp cận để làm phẳng tensor và lặp lại nó.

Có cách nào tốt hơn để tôi có thể lặp qua các hàng, cột và mặt phẳng không?

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

Thư viện Eigen tensor được ghi chép lại ăn dưới cùng của trang này:

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

Từ câu hỏi của bạn, tôi nghĩ rằng bạn sẽ muốn xem xét các hoạt động chip và lát.