Tensorflow：ValueError：不在区间[-1，0]中变暗1。对于“ ExpandDims_1”

Question

我正在尝试为对象检测器设置输入管道。以下是我编写的读取TFrecord的部分代码。

features = tf.parse_single_example(
            serialized_example,
            features=self.features_dict
        )
x_cell = tf.cast(features['x_cell'], tf.int64)
x_cell = tf.expand_dims(x_cell, 1)
return x_cell

当我直接返回x_cell（不执行expand_dims）并通过会话运行它时，我得到了一个大小为（32，）的numpy数组。

当我使用expand_dims时，我希望得到一个数组（32,1）。但是我收到以下错误：

ValueError: dim 1 not in the interval [-1, 0]. for 'ExpandDims_1' (op: 'ExpandDims') with input shapes: [], [] and with computed input tensors: input[1] = <1>.