import tensorflow as tfdataset = 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
tf2
2023-10-13
57 点赞
0 评论
86 浏览