是机器在学习还是亚晨在学习？

tf.py_function 导致 unknown rank

Tensorflow不知道把一个Tensor传入进去之后，你会把什么样子的Tensor传回来，所以我们需要设置一下输出的shape，而且是要在tf.py_funtion外面设置。下面是一个例子：

def echo(x):
    return x
in = tf.constant([[1.0, 2.0], [3.0, 4.0]])
out = tf.py_function(func=echo, inp=[in], Tout=tf.int32)
out.set_shape(in.get_shape())