2016-09-04 14 views
0

でOPSの計算には、未知のディメンションを使用する方法>私は(T == Theano)の作品Theanoコードを、しました:例えばTF

N = input.shape[0] # input has shape wrt TF (?, num) 
mse = T.sum(T.square(ytarg - ypred))/N 

私は本当に、変数Nを使用する方法がわかりませんもしそれが未知の次元であれば、TF演算の計算で。

答えて

1

Tensorflowでは、グラフの実行中に処理する要素の数を知る必要はありません。 tf.reduce_*操作を使用して、このタスクをテンソルフローに委任する必要があります。

削減

TensorFlowあなたはテンソルの様々な寸法を小さく、共通の数学計算を実行するために使用できるいくつかの操作を提供します。

Theanoで定義したMSE関数は、Tensorlowで簡単に定義できます。

mse = tf.reduce_mean(tf.pow(tf.sub(ytarg, ytarg), 2.0)) 
+0

この説明のためのThx! – Glau