具有不同等级的倍增张量

Question

我想将两个不同等级的张量相乘，例如

tensor A -> (i x j x k x l)
tensor B -> (l x m)

获得另一个具有此尺寸的张量。

tensor C -> (i x j x k x m) = (i x j x k x l)*(l x m)

在张量流中，我必须一直重塑以获得输出

(A.reshape(-1,l)*B).reshape(i,j,k,m)

直接进行的任何其他操作？

Answer 1

您需要tf.tensordot。不需要在实数乘法之前重塑。如果要沿A的最后一个维度和B的第一个维度收缩张量A和B，可以按以下方式将张量A和张量B相乘：

tf.tensordot(A, B, axes=[[3], [0]])