tf.TFRecordReader

2022-09-04 20:58:52 浏览数 (5)

从TFRecords文件中输出记录的阅读器。

Eager Compatibility:

读取器与急于执行不兼容,相反,请使用tf,将数据放入模型中。

性能:

实现读取器的Op。

supports_serialize

阅读器实现是否可以序列化其状态。

方法:

__init__

代码语言:javascript复制
__init__(
    name=None,
    options=None
)

创建一个TFRecordReader。

参数:

  • name: 操作的名称(可选)。
  • options: TFRecordOptions对象(可选)

num_records_produced

代码语言:javascript复制
num_records_produced(name=None)

返回此阅读器已生成的记录的数量。这与成功执行的读操作数量相同。

参数:

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

返回值:

  • 一个int64的张量。

num_work_units_completed

代码语言:javascript复制
num_work_units_completed(name=None)

返回此阅读器已完成处理的工作单元的数量。

参数:

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

返回值:

  • 一个int64的张量。

read

代码语言:javascript复制
read(
    queue,
    name=None
)

返回读取器生成的下一个记录(键、值)对。如果需要,将一个工作单元从队列中取出(例如,当读取器需要开始从一个新文件中读取,因为它已经完成了前一个文件)。

参数:

  1. queue: 用字符串工作项表示队列句柄的队列或可变字符串张量。
  2. name: 操作的名称(可选)。

返回值:

  • 张量的元组(键、值)。*键:一个字符串标量张量。值:一个字符串标量张量。

read_up_to

代码语言:javascript复制
read_up_to(
    queue,
    num_records,
    name=None
)

返回最多由读取器生成的num_records(键、值)对。如果需要,将一个工作单元从队列中取出(例如,当读取器需要开始从一个新文件中读取,因为它已经完成了前一个文件)。甚至在最后一个批处理之前,它返回的值也可能小于num_records。

参数:

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

返回值:

  • 张量(键、值)的元组。*键:一维弦张量。*值:一维弦张量。

reset

代码语言:javascript复制
reset(name=None)

将阅读器恢复到其初始干净状态。

参数:

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

返回值:

  • 创建操作。

restore_state

代码语言:javascript复制
restore_state(
    state,
    name=None
)

将阅读器恢复到以前保存的状态。并非所有阅读器都支持恢复,因此这可能会产生未实现的错误。

参数:

  • state: 一个串张量。具有匹配类型的阅读器的串行化的结果。
  • name: 操作的名称(可选)。

返回值:

  • 创建操作。

serialize_state

代码语言:javascript复制
serialize_state(name=None)

生成一个字符串张量来编码阅读器的状态。并不是所有阅读器都支持序列化,因此这可能会产生未实现的错误。

参数:

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

返回值:

  • 一个串张量。

0 人点赞