如何在 Tensorflow 中打印出预测概率

Question

我是 tf. 我已经使用 tensorflow 训练了一个编码器 - 解码器。该程序将一个单词作为输入并打印出它的音素。例如：Hello World -> ['h', 'E', 'l', '"', '@U', ' ', 'w', '"', '3`', 'r', ' 5', 'd']我想获得所选每个音素的预测概率。在预测部分，我使用的代码如下：def predict(words, sess):    if len(words) > hp.batch_size:        after = predict(words[hp.batch_size:], sess)        words = words[:hp.batch_size]    else:        after = []    x = np.zeros((len(words), hp.maxlen), np.int32)  # 0: <PAD>    for i, w in enumerate(words):        for j, g in enumerate((w + "E")[:hp.maxlen]):            x[i][j] = g2idx.get(g, 2)             preds = np.zeros((len(x), hp.maxlen), np.int32)    for j in range(hp.maxlen):        xpreds = sess.run(graph.preds, {graph.x: x, graph.y: preds})        preds[:, j] = xpreds[:, j]提前谢谢你！我的主要问题是这些概率“隐藏”在哪里以及如何访问它们。例如，单词“Hello”中的字母“o”被映射到音素“@U”。我想知道“@U”被选为理想音素的概率是多少。