tensorflow - 张量切片操作中的问题

Question

我试图expected_out从input.

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

如何使用 TensorFlowexpected_out获得。input

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)))