テンソルフローノードのベクトルを効率よく作成する

Question

私は、テンソルフロー変数xといくつかの定数yを取得し、出力としてノードを返します。これは、それらの2つの方法に依存します。

import tensorflow as tf 

x = tf.Variable(3.0) 
y = {"a" : 3, "b" : 1.0} 

def make_graph(x, y): 
    return y["a"] * x**2 + y["b"] 

私はY（y_vec）のような定数のリストを持っていると私は各要素に関数を適用して、これらのノードの合計、このような何かを計算したいと思います：私はその後

f = sum([ make_graph(x, y) for y in y_vec ]) 

をxに関してfを最適化したいもちろん、関数make_graphはもっと複雑になる可能性があります。問題は、非常に長いy_vecに対してこれを効率的に行う方法です。

Answer 1

回答は、適用する機能によって異なります。あなたの例では、次のようなことをすることができます：

import tensorflow as tf 

x = tf.Variable(3.0) 
y = {"a" : [3, 4, 5], "b" : [1.0, 2.0, 3.0]} 

def make_graph(x, y): 
    return tf.reduce_sum(y["a"] * x**2 + y["b"], axis=0) 

f = make_graph(x, y)