Tensorflow Dataset使用

---

前言

tf的版本：2.7.0
记录tensorflow的使用方法

---

一、Dataset使用

# from_tensor_slices：表示从张量中获取数据。
# make_one_shot_iterator():表示只将数据读取一次，然后就抛弃这个数据了。
input_data = [1,2,3,5,8]
dataset = tf.data.Dataset.from_tensor_slices(input_data)
for e in dataset:
    print(e)

二、TextLineDataset，string_split使用,以及中间变量打印

def process(x):
    y = tf.compat.v1.string_split([x], sep=" ") #注意string_split的输入是list
    tf.print(y) #打印中间张量
    return y

dataset = tf.data.TextLineDataset([".\\data\\tf_dataset_text\\1.txt"])
dataset=dataset.map(lambda x : process(x) )
for i in dataset:
    pass