张量繁殖

Question

我有两个大小

a＆lt; tf.tensor'sequent_12/my_layer_56/add：0'shape =（？，300，2）dtype = float32＆gt; gt;

和 b＆lt; tf.tensor'input_82：0'shape =（？，2，2）dtype = float32＆gt; gt;

现在，我想从通常的矩阵行列产品的意义上乘以它们以获得

大小的A * B（？，300，2），因此我只能在第二和第三维中使用矩阵产品。我该如何实现？

我尝试使用具有不同轴规范的TF.Tensordot，但到目前为止尚未工作。例如，我尝试了

tf.tensordot（a，b，axes = [[2]，[0]]），

但这会产生以下形式的张

量2，2）dtype = float32＆gt;

Answer 1

也许尝试tf.matmul：

import tensorflow as tf

samples = 1
A = tf.random.normal((samples, 300, 2))
B = tf.random.normal((samples, 2, 2))

print(tf.matmul(A, B).shape)
# (1, 300, 2)