使用 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

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

您的代码应该与y = tf.expand_dims(x, axis=0). 这将导致[1, 3]您的示例中的形状为，允许您之后平铺。另一种选择是y = x[tf.newaxis, :]它还添加了一个轴。