tf2 数据处理

212 阅读 0 评论 140 点赞

我是靠谱客的博主跳跃电话，这篇文章主要介绍tf2 数据处理，现在分享给大家，希望可以做个参考。

import tensorflow as tf
dataset = tf.data.Dataset.from_tensor_slices(([[1,2],[2,3],[3,4],[1,2],[2,3],[ 3,4]],[4,5,6,4,5,6]))
dataset = dataset.batch(2)
for i in dataset:
print(i)

(<tf.Tensor: id=12, shape=(2, 2), dtype=int32, numpy=
array([[1, 2],
[2, 3]])>, <tf.Tensor: id=13, shape=(2,), dtype=int32, numpy=array([4, 5])>)
(<tf.Tensor: id=14, shape=(2, 2), dtype=int32, numpy=
array([[3, 4],
[1, 2]])>, <tf.Tensor: id=15, shape=(2,), dtype=int32, numpy=array([6, 4])>)
(<tf.Tensor: id=16, shape=(2, 2), dtype=int32, numpy=
array([[2, 3],
[3, 4]])>, <tf.Tensor: id=17, shape=(2,), dtype=int32, numpy=array([5, 6])>)