在 tensorflow 自定义操作中确定内存布局 (ColMajor/RowMajor)。

Question

我写了一个自定义的 tensorflow op 来重塑 Eigen::Tensors 我需要确定输入是行还是列主要对齐。

我尝试转换为 EigenMatrixBase，但没有成功。

Answer 1

特征张量同时支持列和行主要布局。但是，TensorFlow 操作始终使用 Row Major 布局，因此您可以预期输入和输出张量是 Row Major。