2017-09-02 11 views
-1

この質問は、テンソル内の個々の要素、たとえば[1,2,3]にアクセスすることに関してです。私は内側の要素[1]にアクセスする必要があります。私はテンソルフローではない別のAPIライブラリーを呼び出しています。このapiは引数としてint値を必要とします。[1]がintではなくテンソルとして表示されるので、[1]をそのapiに配置する際に問題があります。 [1]のような単一のテンソルをint 1に変換する方法はありますか?テンソルからテンソルにintを取る方法は?

+0

上記のテンソルを作成するために使用したコードを貼り付けてください。 – Tejaswi

答えて

0

セッションを開始し、変数をロード/初期化してからセッションを実行する必要があります。

# build graph 
target_tensor = bigger_tensor[1, 2] # whichever index you want 
with tf.Session() as sess: 
    sess.run(tf.global_variables_initializer()) # or load from file 
    target_value = sess.run(target_tensor) #may need a feed_dict if your model uses placeholders 
+0

ありがとうございます。テンソルフローで損失関数に項を追加しているので、それが私が望む方法であるかどうかは分かりませんが、余分な項は外部パッケージで計算する必要があります。私はそれをセッション内で計算したいと思いますが、私はこの方法を試してみましょう。どうもありがとうございました。 – Joy

+0

トレーニングの途中で損失関数にカスタム演算を追加することを目標とする場合は、別の方法を試してみるとよいでしょう。 'sess.run()'は、同じステップでTensorFlowと非TensorFlowコードを混在させるのではなく、ステップの出力をフェッチするのに本当に便利です。最初にTensorFlow演算子を使用して計算を表現したり、それが機能しない場合は 'tf.py_func'を使用して計算したい場合があります。より詳細なことがなければ、何が正しいのかを言うのは難しいです。 –

関連する問題