AttributeError：模块“张量流”没有属性“选择”

Question

这是一个简单的深度学习程序：

import tensorflow as tf
import numpy as np

A = 3
B = tf.convert_to_tensor([1, 2, 3, 4])
C = tf.convert_to_tensor([1, 1, 1, 1])
D = tf.convert_to_tensor([0, 0, 0, 0])

with tf.Session() as sess:
    print(sess.run(tf.select(A > 1, 'A', 'B')))
    print(sess.run(tf.select(False, 'A', 'B')))
    print(sess.run(tf.select(B > 2, C, D)))

当我运行它时，出现以下错误：

AttributeError: 模块 'tensorflow' 没有属性 'select'。

那么，接下来我该怎么做？有什么函数可以替代 tf.select 吗？

Answer 1

tf.select 自 v0.12 起已弃用，并在 v1.0 中重命名为 tf.where 以匹配 numpy 的命名。 release notes中提到过。