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