tf.assert_rank_at_least() 判断x的秩是不是至少为rank,如果不是抛出异常
参数:
tf.assert_rank_at_least(
x,
rank,
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_at_least(x, 3)]):
output = tf.reduce_sum(x)
print(sess.run(output))