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列

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。