发生乘法的唯一方法是第一个矩阵中的行数与第二个矩阵中的列数匹配。...为了使 和 彼此相乘,必须调换 的第二轴和第三轴。并且两者的大小均为 (3, 3, 2)。这意味着必须变成(3,2,3)。这可以使用 Y.permute(0, 2, 1) 来完成,它转置第二和第三轴。...它还需要第一轴和第二轴与两个张量匹配:
(c、z、m、n) x (c、z、n、r) = (c、z、m、r)
在三维空间中,进行矩阵乘法,然后进行向量之间的点积。...对于本例, 和 的大小为 (2, 3, 3, 2)。为了进行乘法运算,必须调换 的第三轴和第四轴。...这可以按照与之前使用 Y.permute(0, 1, 3, 2) 或 Y.transpose(2,3) 相同的方式完成。转置后的形状为 (2, 3, 2, 3)。