2016-12-12 9 views
0

私はTensorFlowでプレイし始めました。私はtf.nn.softmaxtf.placeholderと連動させることができません。このコードは、TensorFlow - プレースホルダーのsoftmax

import tensorflow as tf 
import numpy as np 

shape = [1, 3] 
value = 0. 
probs = tf.constant(value, shape=shape) 
sampling_prob = tf.nn.softmax(probs) 

with tf.Session() as sess: 
    print(sess.run(sampling_prob)) 

となります。[[ 0.33333334 0.33333334 0.33333334]]を返します。しかし、私がそれを変更すると:

probs2 = tf.placeholder(tf.float32, shape=shape) 
sampling_prob2 = tf.nn.softmax(probs2) 

with tf.Session() as sess: 
    print(sess.run(sampling_prob2, feed_dict={probs2: np.full(shape, value)})) 

突然私は[[ 0. 0. 0.]]を得ます。どのようにそれを可能にすることができますか? Windowsの10、Pythonの3.5.2、TensorFlow 0.12、CUDA 8

+0

Ubuntu 14.04、Python 2.7、TensorFlow 0.11 CPUのみが正しく動作しています – martianwars

答えて

0

コードで実行

は(Ubuntuの、Pythonの2、TensorFlow 0.11 GPU、CUDA 8)私のために正常に動作します。 TensorFlowのWindowsのサポートは、実際には新しく(1ヶ月ほど以内に)、問題の中でこれを言いたいかもしれないようです:https://github.com/tensorflow/tensorflow/issues

関連する問題