2017년 6월 15일 목요일

[TF] tensor to ndarray

https://stackoverflow.com/questions/34097281/how-can-i-convert-a-tensor-into-a-numpy-array-in-tensorflow


predict =   tf.argmax(model, 1)
result =    sess.run(predict , feed_dict = {
    X: x_data,
    Y: y_data,
    keep_prob: 1
    })

print(type(result)) << ndarray
print( 'Predict' , result)

댓글 없음:

댓글 쓰기