在tensorflow自定义op中确定内存布局（ColMajor / RowMajor）。

Question

我已经编写了一个自定义张量流操作，它重塑了Eigen :: Tensors，我需要确定输入是行还是列主要对齐。

我尝试过投射到EigenMatrixBase，但它没有用。

Answer 1

Eigen Tensors支持Column和Row Major布局。但是，TensorFlow操作始终使用Row Major布局，因此您可以预期输入和输出张量为Row Major。