Проблема в операции нарезки тензора

Question

Я пытаюсь получить expected_out от input.

input = [[2],[3],[3]] 
expected_out = [2,3,3] 

Как получить expected_out от input использованием TensorFlow.

Answer 1

В этом случае необходимо удалить одномерные элементы из матрицы. Как в TensorFlow, так и в Numpy эта операция называется squeeze.

Вот эта документация для TensorFlow - tf.squeeze. Цитата из документации,

При использовании тензорного ввода эта операция возвращает тензор того же типа со всеми размерами размера 1, удаленными. Если вы не хотите, чтобы удалить все размеры 1 размеры, вы можете удалить конкретный размер 1 размеры, указав ось

Таким образом, чтобы решить вашу проблему, вы можете либо передать None к axis, ваше дело по умолчанию, или передать 1 , Вот то, что код будет выглядеть,

expected_out = tf.squeeze(input) 

или,

expected_out = tf.squeeze(input, 1) 

Answer 2

tf.squeeze Использование:

import tensorflow as tf 

input = tf.constant([[2], [3], [3]]) 

with tf.Session() as sess: 
    print(sess.run(tf.squeeze(input)))