2016-09-07 16 views
0
Tensor A is [M X 2N X N] 
Tensor B is [M X N] 

は、私が取得するために、2つのテンソルを乗算する必要があります。ここでは はTensorflowのバッチ処理行列の乗算使用

Tensor C [M X 2N X N]. 

は一例です:

M= 2, N = 2 

A: [[[1,2] 
    [1,2] 
    [1,2] 
    [1,2] ] 

    [[2,2] 
    [2,2] 
    [2,2] 
    [2,2] ]] 

B = [[3,3] 
     [2,2]] 

C: [[[3,6] 
     [3,6] 
     [3,6] 
     [3,6] ] 

     [[4,4] 
     [4,4] 
     [4,4] 
     [4,4] ]] 

これを達成する方法がわかりません。その解決策とともに、誰かがこのような状況で放送がどのように正確に機能するかを説明してください。

答えて

0

テンソルBがその次元で2N回繰り返される要素的な乗算が必要なようです。私は、xrange(2 * n)]のtf.concat(2、[tf.reshape(b、[m、1、n])を使って手動でブロードキャストを行います。