Python tf.compat.v1.placeholder_with_default用法及代码示例

一个占位符操作，当其输出未被馈送时通过input。

用法

tf.compat.v1.placeholder_with_default(
    input, shape, name=None
)

迁移到 TF2

警告：这个 API 是为 TensorFlow v1 设计的。继续阅读有关如何从该 API 迁移到本机 TensorFlow v2 等效项的详细信息。见TensorFlow v1 到 TensorFlow v2 迁移指南有关如何迁移其余代码的说明。

强烈建议不要将此 API 用于即刻执行和 tf.function 。此 API 的主要用途是测试包装在 tf.function 中的计算，其中输入张量可能没有静态已知的 fully-defined 形状。通过使用具有部分定义形状的 tf.TensorSpec 输入从 tf.function 创建一个具体函数可以实现相同的目的。例如，代码

@tf.function
def f():
  x = tf.compat.v1.placeholder_with_default(
      tf.constant([[1., 2., 3.], [4., 5., 6.]]), [None, 3])
  y = tf.constant([[1.],[2.], [3.]])
  z = tf.matmul(x, y)
  assert z.shape[0] == None
  assert z.shape[1] == 1

f()

可以很容易地替换为

@tf.function
def f(x):
  y = tf.constant([[1.],[2.], [3.]])
  z = tf.matmul(x, y)
  assert z.shape[0] == None
  assert z.shape[1] == 1

g = f.get_concrete_function(tf.TensorSpec([None, 3]))

您可以在使用 tf.function 获得更好的性能中了解有关 tf.function 的更多信息。