如何打印tensorflow 的dataset

在 TensorFlow 中,为了调试数据,通常需要打印dataset的内容。由于直接打印可能会遇到错误,例如因尝试创建一个不支持状态对象(如Variable或LookupTable)的一次性迭代器而报错。解决这个问题的方法是使用`Dataset.make_initializable_iterator()`代替`Dataset.make_one_shot_iterator()`。
摘要由CSDN通过智能技术生成

有时,为了调试数据,需要将数据打印打出来,可以用interator来遍历数据

 

首先定义两个遍历函数,

    def print_dataset(self, data_set):
        iterator = data_set.make_one_shot_iterator()
        next_element = iterator.get_next()
        num_batch = 0
        with tf.train.MonitoredTrainingSession() as sess:
            while not sess.should_stop():
                value = sess.run(next_element)
                num_batch += 1
                print("
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值