使用 expand_dims 时，“尝试将 'dim' 转换为张量但失败了”

Question

我希望下面的代码产生相应的张量

>>> A = tf.range(3)
>>> B = tf.tile(tf.expand_dims(A), [4,1])
>>> print(tf.Session().run(B))

[[0,1,2],
 [0,1,2],
 [0,1,2],
 [0,1,2]]

但是这会导致ValueError。重现问题的简单方法如下

>>> import tensorflow as tf
>>> x = tf.constant([0,1,2])
>>> y = tf.expand_dims(x)

ValueError: Tried to convert 'dim' to a tensor and failed. Error: None values not supported.

使用expand_dims 并避免此错误的正确方法是什么？

Answer 1

看起来您需要为expand_dims 的axis 参数指定一个值。默认值为None，这似乎会导致您遇到错误。这有点奇怪，因为默认参数通常会导致某种合理的默认行为......也许这是一个错误。

您的代码应该适用于y = tf.expand_dims(x, axis=0)。这将在您的示例中生成[1, 3] 的形状，允许您在之后进行平铺。另一个选项是y = x[tf.newaxis, :]，它还添加了一个轴。