TensorFlow函数:tf.WholeFileReader

2018-04-24 10:28 更新

tf.WholeFileReader函数

WholeFileReader类

继承自:ReaderBase

定义在:tensorflow/python/ops/io_ops.py.

请参阅指南:输入和读取器>读取器

将文件的全部内容作为值输出的Reader.

如果要使用,请在队列(Queue)中的排列文件名.Read的输出将是一个文件名(key)和该文件的内容(value).

有关支持的方法,请参阅ReaderBase.

Eager兼容性

读者与eager的执行不兼容.相反的,你需要使用tf.data将数据存入您的模型.

属性

  • reader_ref
    实现了读取器的操作.
  • supports_serialize
    表示Reader实现是否可以序列化其状态.

方法

__init__

__init__(name=None)

该方法用于创建一个WholeFileReader.

参数:

  • name:操作的名称(可选).

num_records_produced

num_records_produced(name=None)

返回此读取器生成的记录数.

这与已成功读取执行的次数相同.

参数:

  • name:操作的名称(可选).

返回值:

该方法返回一个类型为int64的张量.

num_work_units_completed

num_work_units_completed(name=None)

返回此读取器完成处理的工作单位数.

参数:

  • name:操作的名称(可选).

返回值:

该函数返回一个类型为int64的张量.

read

read(
    queue,
    name=None
)

返回读取器生成的下一条记录(key,value)对.

如果需要,将从队列中取出一个工作单元(例如,当Reader需要从一个新文件开始读取,因为它已经完成了前一个文件).

参数:

  • queue:队列或可变字符串张量,表示队列的句柄,包含字符串工作项.
  • name:操作的名称(可选).

返回值:

张量元组(key,value),其中,key是一个字符串标量张量,value是一个字符串标量张量.

read_up_to

read_up_to(
    queue,
    num_records,
    name=None
)

最多返回由读取器生成的num_records(key,value)对.

如果需要,将从队列中取出一个工作单元(例如,Reader需要从一个新文件开始读取,因为它已经完成了前一个文件).即使在最后一批之前,它的返回率也可能少于num_records.

参数:

  • queue:队列或可变字符串张量,表示队列的句柄,包含字符串工作项.
  • num_records:要读取的记录数.
  • name:操作的名称(可选).

返回值:

张量元组(keys,values),其中,keys是一维字符串张量,values也是一维字符串张量.

reset

reset(name=None)

将读取器恢复到其初始状态.

参数:

  • name:操作的名称(可选).

返回值:

该方法返回创建的操作.

restore_state

restore_state(
    state,
    name=None
)

将读取器恢复到先前保存的状态.

并非所有读取器都能够支持restore,所以这可能会产生未实现的错误.

参数:

  • state:一个字符串张量,具有匹配类型的Reader的SerializeState的结果.
  • name:操作的名称(可选).

返回值:

该方法返回创建的操作.

serialize_state

serialize_state(name=None)

生成一个字符串张量,用于编码读取器的状态.

并非所有的读取器都支持序列化,所以这可能会产生未实现的错误.

参数:

  • name:操作的名称(可选).

返回值:

该方法返回一个字符串张量.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号