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

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

Eigen Tensor库记录在本页底部：

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

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