如何迭代张量的行,列和平面？

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

从你的问题来看，我认为你会想看看芯片和切片操作。