tensorflow - Tensor スライス操作の問題

Question

expected_outから取得しようとしていますinput。

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

TensorFlow を使用してexpected_outfromを取得する方法。input

Answer 1

tf.squeezeを使用します:

import tensorflow as tf

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

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