python - Tensorflow:Py_func 返回未知形状

Question

我有一个关于 tf.py_func 函数的简单问题。

我有一个形状为 (1,224,224,3) 的图像张量 my_img。为了测试 py_func，我将张量提供给 python 函数 return_tf，该函数应该返回相同的张量(在根据文档转换为 numpy 数组之后)。

代码如下:

def return_tf(x):
   return np.array(x)

test = tf.py_func(return_tf,[my_img],[tf.float32])

但是当我检查名为 test 的返回张量的形状时，我得到:

tf.Tensor 'PyFunc:0' shape=unknown dtype=float32

我也无法在张量上运行 eval()，因为出现错误:

AttributeError: 'list' object has no attribute 'eval'.

任何人都知道如何修复 tf.py_func 返回的张量的张量形状？

Answer 1

刚刚找到了一个解决方法..因为 py_func 返回一个 tensorflow 列表，我可以做 ff:

test = tf.reshape(tf.concat(1, test), [ <<a shape>> ])

获得具有所需形状的张量