Python 在tensorflow中实现简单线性代数运算

给定一些列为X_1，…，X_m的（n，m）矩阵X，我试图找到一个运算，它给出了3模张量[X_1 X_1^T，…，X_m X_m^T]，其中vec是矩阵X_I X_I^T的向量化

换言之，我试图概括

tf.tensordot(a,a,axes=0)

或

从向量a到矩阵的列。有没有一种方法可以在不依赖循环的情况下完成此操作？

您可以通过以下方式完成此操作：

tf.expand_dims(a, 2) @ tf.expand_dims(a, 1)

或者，如果愿意，可以使用而不是

@

运算符。

您可以使用以下方法：

tf.expand_dims(a, 2) @ tf.expand_dims(a, 1)

---

或者如果您愿意，可以使用而不是

@

运算符。

谢谢。这样可以计算行的张量积，但也可以，谢谢。这样，它将计算行的张量积，但也可以。