标签: tensorflow matrix-multiplication linear
让a为2D张量定义为:
a
a = tf.constant([[1,2,3], [1,2,3], [1,2,3]], dtype=tf.float32)
如何将每列j乘以索引为>j的其余列?应用于a的此操作将产生以下结果:
j
>j
[[2,3,6], [2,3,6], [2,3,6]]
我已经发布了一个带有该问题here变体的问题。不幸的是,我找不到为此使用tf.einsum的方法。
tf.einsum