Python tf.io.read_file用法及代码示例

读取文件的内容。

用法

tf.io.read_file(
    filename, name=None
)

此操作返回一个张量，其中包含输入文件名的全部内容。它不做任何解析，它只是按原样返回内容。通常，这是输入管道的第一步。

例子：

with open("/tmp/file.txt", "w") as f:
  f.write("asdf")

4
tf.io.read_file("/tmp/file.txt")
<tf.Tensor:shape=(), dtype=string, numpy=b'asdf'>

在函数中使用 op 读取图像、对其进行解码并重塑包含像素数据的张量的示例：

@tf.function
def load_image(filename):
  raw = tf.io.read_file(filename)
  image = tf.image.decode_png(raw, channels=3)
  # the `print` executes during tracing.
  print("Initial shape:", image.shape)
  image.set_shape([28, 28, 3])
  print("Final shape:", image.shape)
  return image