利用tf.data.Dataset API读取numpy array文件
读取numpy array数据
import tensorflow as tf
import numpy as np
from sklearn import datasets
iris = datasets.load_iris()
ds1 = tf.data.Dataset.from_tensor_slices((iris['data'],iris['target']))
for feature, label in ds1