當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.debugging.Assert用法及代碼示例


斷言給定條件為真。

用法

tf.debugging.Assert(
    condition, data, summarize=None, name=None
)

參數

  • condition 要評估的條件。
  • data 當條件為假時要打印的張量。
  • summarize 打印每個張量的這麽多條目。
  • name 此操作的名稱(可選)。

返回

拋出

如果 condition 評估為 false,則在 data 中打印張量列表。 summarize 確定要打印的張量條目數。

注意:應該使用這個函數的輸出。如果不是,則會記錄警告或引發錯誤。要將輸出標記為已使用,請調用其 .mark_used() 方法。

TF1 兼容性

當處於 TF V1 模式時(即在 tf.function 之外),斷言需要對輸出的控製依賴以確保斷言執行:

# Ensure maximum element of x is smaller or equal to 1
assert_op = tf.Assert(tf.less_equal(tf.reduce_max(x), 1.), [x])
with tf.control_dependencies([assert_op]):
  ... code using x ...

eager模式兼容性

返回None。

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.debugging.Assert。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。