tf2 数据处理

小步想当程序员

于 2020-08-31 17:44:41 发布

阅读量355

点赞数

文章标签： tf2

本文链接：https://blog.youkuaiyun.com/qq_41175067/article/details/108326197

版权

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