给定两个NxD矩阵，以更有效的实现方式生成NxDxD矩阵

Question

我有两个形状为A的矩阵(N,D)和形状为B的{​​{1}} 对于A [i]和B [i]（均为D维向量）的每一行，我正在通过矩阵乘法构造DxD矩阵。最后，我将得到NxDxD张量

现在；我的代码效率不高，需要循环；有什么快速的方法吗？

(N,D)

Answer 1

从本质上讲，您的x或y操作实际上是矩阵N与矩阵Dx1的{​​{1}}的矩阵相乘，每对都会给您一个N矩阵。这样就可以做到：

1xD

一个例子：

DxD