1

3つのスパンテンソルのA = P * N、B = Q * N、C = R * Nがあります。積行列A * B * Cを計算する効率的な方法は何ですか。その結果、積行列の次元はテンソルフローでP * Q * Rになります。テンソルフローのテンソル連鎖を効率的に乗算する方法はありますか?

私はtf.matmulを試して、tf.reshapeを続けましたが、上記で指定された次元の製品行列を与えません。

ありがとうございました。

答えて

0

tf.einsumお手伝いが必要です。 A*B*Cの意味がわからないのですが、次元は行列の乗算と互換性がないためです。おそらく次のようなものでしょう:

R = tf.einsum('il,jl,kl->ijk', A, B, C) 
関連する問題