在Tensorflow中进行操作时如何设置可变的输出形状？

Question

在Tensorflow中进行操作期间，我必须在用REGISTER_OP进行注册操作期间设置输出的形状。

具体来说，我使用SetShapeFn来设置输出。但是，如果无法固定输出的形状，则不知道如何设置输出的形状。

直到现在，如果无法固定输出形状，我仍将输出的形状设置为UnknownShape，但是我担心运行此操作期间的一些缺点。

如果将输出形状设置为UnknownShape，会不会有不好的效果？

我想获得一些建议。

谢谢。

Answer 1

某些op需要特定的输入形状，例如tf.image.resize_bilinear要求输入形状为4维。 如果像这样的op在您的op之后，则会发生错误。

一种解决方法是，可以在操作之后使用tf.reshape来定义特定的输出形状。