减少张量流中张量的尺寸[重复]

Question

我有一个形状如下的张量：

> tf.Tensor: shape=(1, 1440)

如何减少这样的形状，以便得到以下结果：

> tf.Tensor: shape=(1440,)

Answer 1

使用tf.squeeze:

import tensorflow as tf

tensor = tf.random.uniform((1, 1440))
print(tensor.shape

TensorShape([1, 1440])

现在：

squeezed = tf.squeeze(tensor)
print(squeezed.shape)

TensorShape([1440])

如果您真的想要形状的逗号格式，请将其转换为 NumPy：

tf.squeeze(s).numpy().shape

(1440,)