TensorFlow：高维张量的加权和？

Question

假设我得到了一个形状为for (var prop in Traits) { if(Traits[prop] & 6) { console.log(prop); } } 的三维张量input。

如果我的(batch_size, dim_a, dim_b)向量的形状为weight，并希望得到形状为(batch_size, dim_a)的{​​{1}}输入的加权和。我该怎么办？

Answer 1

您可以重塑您的体重向量并使用SELECT (SELECT SUM(myColumnOne) from myTableOne) / (SELECT SUM(myColumnTwo) FROM myTableTwo) * 100 AS myResult; ：

tf.reduce_sum