ARGMAXを

Question

私はその値が同様であることをTensorflowにおけるテンソルがあるとします。他では

B = [[1, 0, 0],[0, 0, 1]] 

：私は次のように、このテンソルを変更するにはどうすればよい

A = [[0.7, 0.2, 0.1],[0.1, 0.4, 0.5]] 

を言葉を最大限にして1に置き換えたい。
助けてください。

Answer 1

私はあなたがワンライナーでそれを解決することができると思います。

import tensorflow as tf 
import numpy as np 

x_data = [[0.7, 0.2, 0.1],[0.1, 0.4, 0.5]] 
# I am using hard-coded dimensions for simplicity 
x = tf.placeholder(dtype=tf.float32, name="x", shape=(2,3)) 

session = tf.InteractiveSession() 

session.run(tf.one_hot(tf.argmax(x, 1), 3), {x: x_data}) 

結果はあなたが期待するものです。

Out[6]: 
array([[ 1., 0., 0.], 
     [ 0., 0., 1.]], dtype=float32)