如何在tensor的前方增加维度

你可以使用 TensorFlow 中的 tf.expand_dims() 函数来在张量的前方添加一个新维度。该函数需要两个参数：要扩展的张量和新维度的索引。例如，以下代码将在张量 x 的前方添加一个新维度：

import tensorflow as tf

x = tf.constant([1, 2, 3, 4])
x = tf.expand_dims(x, 0)
print(x.shape) # (1, 4)

在此示例中，我们将张量 x 扩展到第 0 个维度，因此它现在具有形状 (1, 4)。注意，新维度的大小必须为 1。