tf.squeeze

tf.squeeze(input, squeeze_dims=None, name=None)

Removes dimensions of size 1 from the shape of a tensor.
从tensor中删除所有大小是1的维度

输入：张量。 输入要挤压。
squeeze_dims：可选的ints列表。 默认为[]。 如果指定，只能挤压列出的尺寸。 维度索引从0开始。挤压不是1的维度是一个错误。

示例：
print(“sampled_indices1:”,sampled_indices)
sampled_indices = tf.squeeze(sampled_indices,axis=-1)
print(“sampled_indices2:”,sampled_indices)

输出：
sampled_indices1: tf.Tensor(
[[26]
[46]
…
[27]
[12]], shape=(100, 1), dtype=int64)
sampled_indices2: tf.Tensor(
[26 46 … 27 12], shape=(100,), dtype=int64)