Output from TensorFlow `py_func` has unknown rank/shape

Since py_func can execute arbitrary Python code and output anything, TensorFlow can’t figure out the shape (it would require analyzing Python code of function body) You can instead give the shape manually

y.set_shape(inp.get_shape())