TensorFlow Eager模式--数据导入(tf.data)

在Eager Execution模式下,placeholder操作无法与之兼容,可以使用tf.data提供的API来导入数据

Eager模式简单教程https://github.com/madalinabuzau/tensorflow-eager-tutorials

Eager模式下使用tf.data和静态图模式下基本一致,区别只在于迭代的创建。在未来,或许 Google会消除这些不同。

一. Eager模式

开启Eager模式的方法,前面的文章已经介绍了2种,最初的eager模块api放在contrib中,现在新版的TensorFlow中已成为核心API,可以直接访问

二. Eager 模式下的tf.data

#eager 在 contrib 中时的开启方法
import tensorflow.contrib.eager as tfe
tfe.enable_eager_execution()

dataset = tf.data.Dataset.from_tensor_slices(np.array([1.0, 2.0, 3.0, 4.0, 5.0]))

for one_element in tfe.Iterator(dataset): # 注意区别
    #print(one_element)
    print(one_element.numpy(), " -> ", one_element)

或者

#eager 成为 核心api 后的开启方法
import tensorflow as tf
tf.enable_eager_execution()

dataset = tf.data.Dataset.from_tensor_slices(np.array([1.0, 2.0, 3.0, 4.0, 5.0]))

for one_element in tf.data.Iterator(dataset): # 注意区别
    print(one_element.numpy(), " -> ", one_element)

输出

1.0  ->  tf.Tensor(1.0, shape=(), dtype=float64)
2.0  ->  tf.Tensor(2.0, shape=(), dtype=float64)
3.0  ->  tf.Tensor(3.0, shape=(), dtype=float64)
4.0  ->  tf.Tensor(4.0, shape=(), dtype=float64)
5.0  ->  tf.Tensor(5.0, shape=(), dtype=float64)

tensorflow==1.9

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值