tensorflow 사용자 정의 연산에서 메모리 레이아웃 (ColMajor/RowMajor)을 결정하십시오.

Question

나는 Eigen :: Tensors를 재구성하는 사용자 정의 tensorflow 연산을 작성했으며 입력 값이 행 또는 열의 길이가 맞는지 확인해야합니다.

EigenMatrixBase로 캐스팅을 시도했지만 작동하지 않았습니다.

Answer 1

고유 텐서는 열 및 행 주요 레이아웃을 모두 지원합니다. 그러나 TensorFlow 작업은 항상 행 메이저 레이아웃을 사용하므로 입력 및 출력 텐서가 행 메이저가 될 것으로 예상 할 수 있습니다.