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

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


断言 x 的排名等于 rank 。

用法

tf.compat.v1.assert_rank(
    x, rank, data=None, summarize=None, message=None, name=None
)

参数

  • x 数字 Tensor 。
  • rank 标量整数 Tensor 。
  • data 如果条件为 False，则打印出的张量。默认为错误消息和 x 的形状。
  • summarize 打印每个张量的这么多条目。
  • message 默认消息的前缀字符串。
  • name 此操作的名称(可选)。默认为"assert_rank"。

返回

  • 除非 x 已指定排名，否则操作会提高 InvalidArgumentError。如果静态检查确定x 的排名正确，则返回no_op。

抛出

  • ValueError 如果静态检查确定x 排名错误。

将依赖项添加到操作的示例：

with tf.control_dependencies([tf.compat.v1.assert_rank(x, 2)]):
  output = tf.reduce_sum(x)

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.assert_rank。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。