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 的更多信息。