如何使用TensorFlow将每个张量列乘以同一张量的其余列?

时间:2018-06-19 18:29:03

标签: tensorflow matrix-multiplication linear

a为2D张量定义为:

a = tf.constant([[1,2,3], 
                 [1,2,3], 
                 [1,2,3]], dtype=tf.float32)

如何将每列j乘以索引为>j的其余列?应用于a的此操作将产生以下结果:

[[2,3,6],
 [2,3,6],
 [2,3,6]]

我已经发布了一个带有该问题here变体的问题。不幸的是,我找不到为此使用tf.einsum的方法。

0 个答案:

没有答案