Добавление дополнительного измерения с размером 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)])