形状が[batch_size, height, width, depth]
、テンソルが[depth]
の層の出力を持つ場合、どのようにしてdepth
に沿った各スライスに2番目のテンソルの対応する値が乗算されるように、最初のテンソルに2番目のテンソルを乗算できますか?二テンソルが[4, 5, 6]
されれば、その後、乗算は次のとおりです。任意の軸に沿って掛け合っていますか?
tensor1[:, :, :, 0] * 4
tensor1[:, :, :, 1] * 5
tensor1[:, :, :, 2] * 6
また、私が検索するために知らなかった乗算のこの種の名前はありますか?ありがとう!
正しい軸が乗算に使用されていることを確認するだけで、 'depth 'を' [1,1,1,3] 'に変更できます。 ( 'tf.expand_dims()'も使用できます) –
はい、私は同意します。私の例では放送が暗黙的なので、 'depth = tf.reshape(tf.constant([4,5,6]、dtype = tf.float32)、shape = [1,1,1,3]より明確になります。 – rvinas