Добавление информации в тензор в TensorFlow

Question

В настоящее время у меня есть нейронная сеть, которая свертывает и объединяет изображения. Однако, прежде чем я сделаю свой плотно связанный слой, я хочу добавить некоторую информацию. В настоящее время я изменяю свое изображение на плоский тензор с tf.reshape(image, [width * length * channels]), но мне было интересно, как я могу добавить пару значений tf.float32 в конец тензора?

Answer 1

Вы можете изменить к вектору, используя tf.reshape с -1:

tf.reshape(image, [-1]) 

и приобщать новые значения, как тензоры с помощью tf.concat:

tf.concat(0, [image, new_val1, new_val2]) 

Это вернет тензорный в результате конкатенации ввода тензоры.