python - tensorflow 将每一行除以不同的总和

Question

假设我有一个这样的张量:

T = [[1,2,3,4], 
    [5,6,7,8], 
    [9,10,11,12]]. 

我已经使用 reduce_sum(T, 1) 获得不同张量中的所有总和，但我似乎无法找到一种方法来实际按行划分总和。我如何在 Tensorflow 中将每一行除以它的总和(即将 [1,2,3,4] 除以 10，[5,6,7,8] 除以 26，等等)？

Answer 1

这很容易实现，只需将矩阵元素除以表示每个维度总和的向量即可:

import tensorflow as tf
a = tf.constant([
    [1, 2, 3, 4],
    [5, 6, 7, 8]
], dtype=tf.float32)

b = tf.reduce_sum(a, axis=1)

c = a / tf.reshape(b, (-1, 1))
with tf.Session() as sess:
    print sess.run(c)