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

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


斷言給定的Tensor 是指定的類型。

用法

tf.debugging.assert_type(
    tensor, tf_type, message=None, name=None
)

參數

  • tensor Tensor , SparseTensor 或 tf.Variable .
  • tf_type 張量流類型(dtypes.float32,tf.int64,dtypes.bool， 等等)。
  • message 默認消息的前綴字符串。
  • name 此操作的名稱。默認為"assert_type"

拋出

  • TypeError 如果張量的數據類型與 tf_type 不匹配。

這始終可以靜態檢查，因此此方法不返回任何內容。

例子：

a = tf.Variable(1.0)
tf.debugging.assert_type(a, tf_type= tf.float32)
b = tf.constant(21)
tf.debugging.assert_type(b, tf_type=tf.bool)
Traceback (most recent call last):

TypeError:...
c = tf.SparseTensor(indices=[[0, 0], [1, 2]], values=[1, 2],
 dense_shape=[3, 4])
tf.debugging.assert_type(c, tf_type= tf.int32)

相關用法


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