但是,第一个轴必须相同:
(z, m, n) x (z, n, r) = (z, m, r)
为什么是这样?嗯,如前所述,二维的点积主要是将向量彼此相乘。...为了使 和 彼此相乘,必须调换 的第二轴和第三轴。并且两者的大小均为 (3, 3, 2)。这意味着必须变成(3,2,3)。这可以使用 Y.permute(0, 2, 1) 来完成,它转置第二和第三轴。...[ 1, 0]],
[[ 1, 1],
[ 6, -5],
[ 8, -5]]]])
点积
在四维中,张量乘法将具有与三维和二维中相同的要求...相同的步骤将在四个维度中发生,但首先将每个 3D 张量与其相应的 3D 张量相乘。然后,它们的每个矩阵将相互相乘。最后,它们的向量将相互执行点积。这可以在上图中看到。...对于本例, 和 的大小为 (2, 3, 3, 2)。为了进行乘法运算,必须调换 的第三轴和第四轴。