Question

我的问题如下：

当我想用列轴中的向量划分矩阵时，但是当A是具有[n,n]形状的矩阵且v带有{[1,n]的向量时，TensorFlow仅提供行划分。 1}}形状。

我的解决方案是：

tf.transpose(tf.div(tf.transpose(A), v))

~~我试过了，但它不起作用：~~ 更新：它有效！

tf.div(A, tf.transpose(v))

所以我的问题是，如果存在更有效的解决方案来解决这个问题。

Answer 1

使用A / tf.transpose(v)的第二个解决方案应该可行。 tf.div()没有轴参数，因为它＆＃34;将x / y元素划分为＆＃34; 。所以两个张量都应该具有相同的尺寸。

在您的情况下，它适用于不同的维度，因为tf.div supports broadcasting。有关广播的更多信息是here。

它也尽可能高效，所以不需要寻找其他任何东西。