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

确定tensorflow自定义操作中的内存布局(ColMajor/RowMajor)。,tensorflow,eigen,Tensorflow,Eigen,我已经编写了一个定制的tensorflow op,它重塑了Eigen::Tensors的形状,我需要确定输入是行对齐还是列对齐 我试着对MatrixBase进行强制转换,但没有成功 本征张量支持列和行的主要布局。但是,TensorFlow操作始终使用行主布局,因此可以预期输入和输出张量为行主布局。特征张量支持列主布局和行主布局。但是,TensorFlow操作始终使用行主布局,因此可以预期输入和输出张量为行主张量。张量应为所有行主顺序张量应为所有行主顺序张量

我已经编写了一个定制的tensorflow op,它重塑了Eigen::Tensors的形状,我需要确定输入是行对齐还是列对齐


我试着对MatrixBase进行强制转换,但没有成功

本征张量支持列和行的主要布局。但是,TensorFlow操作始终使用行主布局,因此可以预期输入和输出张量为行主布局。

特征张量支持列主布局和行主布局。但是,TensorFlow操作始终使用行主布局,因此可以预期输入和输出张量为行主张量。

张量应为所有行主顺序张量应为所有行主顺序张量