TF.DATA 模塊

tf.data模塊簡介

import tensorflow as tf # 使用一個列表建立dataset dataset = tf.data.Dataset.from_tensor_slices([1,2,3,4,5]) dataset

# 應用dataset for ele in dataset:print(ele)

# 把Tensor輸出為numpy類型 for ele in dataset:print(ele.numpy())

# 使用一個嵌套列表建立dataset dataset = tf.data.Dataset.from_tensor_slices([[1,2,3,],[4,5,6],[7,8,9]]) dataset

for ele in dataset:print(ele)

for ele in dataset:print(ele.numpy())

dataset_dic = tf.data.Dataset.from_tensor_slices({"A":[1,2,3,4],"B":[5,6,7,8],"C":[10,11,12,13]}) dataset_dic

for ele in dataset_dic:print(ele)

import numpy as np dataset = tf.data.Dataset.from_tensor_slices(np.array([1,2,3,4,5])) for ele in dataset:print(ele.numpy())

# 取值 dataset = tf.data.Dataset.from_tensor_slices([1,2,3,4,5,6,7,8,9]) for ele in dataset.take(4):print(ele.numpy())

next(iter(dataset.take(1)))

# shuffle 亂序dataset dataset = dataset.shuffle(9)

# repeat循環count=3重復3次 dataset = dataset.repeat(count=3) for ele in dataset:print(ele.numpy())

# batch 每一次請求出來3個數字 dataset = dataset.batch(3) for ele in dataset:print(ele.numpy())

# 總結 dataset = tf.data.Dataset.from_tensor_slices([1,2,3,4,5]) # shuffle 亂序dataset dataset = dataset.shuffle(5) # repeat循環count=3重復3次 dataset = dataset.repeat(count=3) # batch 每一次請求出來3個數字 dataset = dataset.batch(3) # repeat循環count=3重復3次 dataset = dataset.repeat(count=3) for ele in dataset:print(ele.numpy())

# tf.square做平方運算 dataset = tf.data.Dataset.from_tensor_slices([1,2,3,4,5]) dataset = dataset.map(tf.square) for ele in dataset:print(ele.numpy())

總結

以上是生活随笔為你收集整理的深度学习-Tensorflow2.2-tf.data输入模块{2}-tf.data基础用法-09的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。