当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。