TensorFlow 2.0 补充

x = tf.linspace(-5.0,5,11)   # 生成线性均分数据集 tf.Tensor([-5. -4. -3. -2. -1. 0. 1. 2. 3. 4. 5.], shape=(11,), dtype=float32)

train_db = tf.data.Dataset.from_tensor_slices((train_dataset.values,train_labels.values))# 构建 Dataset 对象

train_db = train_db.shuffle(100).batch(32)# 随机打散，批量化

a,b=train_x.take(range(13),axis=1),train_x.take(range(13,39),axis=1)

a：0到13列；b:第13到39列