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])>)