向张量添加大小为None的额外尺寸

Question

如何为张量添加None外部尺寸？

具体来说，我有一个张量为[299,299]的张量，希望作为张量流SavedModel对象的输出返回。但是SavedModel对象希望外部尺寸未知，以便能够处理批次。

tf.expand_dims()添加了一个额外的尺寸，但它设置为1。

Answer 1

In [1]: import tensorflow as tf    
        x = tf.constant([3., 2.])

您可以为此使用tf.reshape()

In [1]: tf.reshape(x, [2, 1])
Out[1]: TensorShape([Dimension(2), Dimension(1)])