tf.data.Dataset.from_tensor_slices作用

该函数是dataset核心函数之一,它的作用是把给定的元组、列表和张量等数据进行特征切片。切片的范围是从最外层维度开始的。如果有多个特征进行组合,那么一次切片是把每个组合的最外维度的数据切开,分成一组一组的。

import tensorflow as tf
    import numpy as np
    
    features, labels = (np.random.sample((6,3)),np.random.sample((6,1)))# 模拟6组数据,每组数据3个特征
print((features, labels)) data = tf.data.Dataset.from_tensor_slices((features,labels)) print(list(data.as_numpy_iterator()))

自行打印

原文地址:https://www.cnblogs.com/moomcake/p/15463488.html