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)