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)

发布了21 篇原创文章 · 获赞 18 · 访问量 1458

猜你喜欢

转载自blog.csdn.net/zephyr_wang/article/details/104061318
tf
今日推荐