这导致:
(m, n) x (n, r) = (m, r)
如果情况并非如此,则必须转置其中一个矩阵以适应该顺序;这会切换行和列,但保留点积的向量。...但是,第一个轴必须相同:
(z, m, n) x (z, n, r) = (z, m, r)
为什么是这样?嗯,如前所述,二维的点积主要是将向量彼此相乘。...为了使 和 彼此相乘,必须调换 的第二轴和第三轴。并且两者的大小均为 (3, 3, 2)。这意味着必须变成(3,2,3)。这可以使用 Y.permute(0, 2, 1) 来完成,它转置第二和第三轴。...[[ 1, 1],
[ 6, -5],
[ 8, -5]]]])
点积
在四维中,张量乘法将具有与三维和二维中相同的要求。...它还需要第一轴和第二轴与两个张量匹配:
(c、z、m、n) x (c、z、n、r) = (c、z、m、r)
在三维空间中,进行矩阵乘法,然后进行向量之间的点积。