tf.assert_rank_in（）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fyq201749/article/details/82191762

tf.assert_rank_in（） 判断x的秩是否在ranks之间，如果不是抛出异常
参数：

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

使用案例：

import tensorflow as tf
import numpy as np
x = [[1, 2, 3], [4, 5, 6]]
sess = tf.Session()
with tf.control_dependencies([tf.assert_rank_in(x=x, rank=(2, 4))]):
  output = tf.reduce_sum(x)
  print(sess.run(output))