ValueError: Rank mismatch in 'train_data/ReadFile' (op: ReadFile)

阅读量：

使用函数tf.train.slice_input_producer读取文件时,

复制代码

    input_queue = tf.train.slice_input_producer(
            [flist], shuffle=self.shuffle,
            seed=0123, num_epochs=self.num_epochs)
    input_file = tf.read_file(input_queue)

出现错误:

InvalidArgumentError: The dimensions must be rank 0 rather than rank 1 for the operation at path 'train_data/ReadFile' (op: ReadFile) when processing input shapes [1].

该函数tf.train.slice_input_producer()返回input_queue的类型是列表。

复制代码

    filename: A `Tensor` of type `string`.

因此需要将:

复制代码

    input_file = tf.read_file(input_queue)

修改为:

复制代码

inp

全部评论 (0)

还没有任何评论哟~

是否确定退出登录?

ValueError: Rank mismatch in 'train_data/ReadFile' (op: ReadFile)

全部评论 (0)

相关文章推荐