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

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


断言 x 在 ranks 中有排名。

用法

tf.compat.v1.assert_rank_in(
    x, ranks, data=None, summarize=None, message=None, name=None
)

参数

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

返回

  • 操作提高 InvalidArgumentError 除非 x 的排名在 ranks 中。如果静态检查确定 x 具有匹配的排名，则返回 no_op。

抛出

  • ValueError 如果静态检查确定x 排名不匹配。

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

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

相关用法


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