如何在TensorFlow中将每行中的最大值更改为1，将所有其他数字更改为0？

import tensorflow as tf

# 创建一个示例矩阵
matrix = tf.constant([[1, 2, 3],
                     [4, 5, 6],
                     [7, 8, 9]])

# 找到每行的最大值的索引
max_indices = tf.argmax(matrix, axis=1)

# 将索引转换为独热编码
one_hot_matrix = tf.one_hot(max_indices, depth=tf.shape(matrix)[1])

# 找到每行的最大值
max_values = tf.reduce_max(matrix, axis=1, keepdims=True)

# 将最大值替换为1，其他数字替换为0
result = tf.where(tf.equal(matrix, max_values), tf.ones_like(matrix), tf.zeros_like(matrix))

# 打印结果
with tf.Session() as sess:
    print(sess.run(result))

[[0 0 1]
 [0 0 1]
 [0 0 1]]