희소 행렬 대각 승산 tensorflow

Question

I 대칭 희소 행렬 M 대각선 행렬 D. 제가

M은 SparseTensor로 만들어 DMD를 계산하려면, 및 D가 tf.diag으로 또는 어느 표현 될 수있다 SparseTensor - 이론적으로 밀집한 매트릭스를 거치지 않고 DMD를 평가하는 것이 전적으로 가능해야하지만, 어떻게 알아낼 수 없는지 - 누군가에게 조언이 있습니까?

Answer 1

다음은 몇 가지 팁입니다.

M 앞뒤에 D를 곱하는 효과는 D (i, i) 및 D (j, j)로 각각의 0이 아닌 요소 M (i, j)를 곱하는 것임을 유의하십시오. here을 참조하십시오.

말 P = DMD. sparse tensor으로 P를 구성하려면 인덱스, 값 및 모양이 필요합니다. M의 인덱스와 모양으로 인덱스와 모양을 설정하고 M의 값과 D (D는 Dense 1D Tensor 일 수 있음)보다 tf.map_fn의 값을 계산할 수 있습니다.