首页 > 其他 > 详细

paddle数据Tensor

时间:2020-05-20 00:11:11      阅读:155      评论:0      收藏:0      [点我收藏+]

1.根据输入数据构建loader

  

src = fluid.data(name="src", shape=[None, None], dtype="int64")
    src_sequence_length = fluid.data(name="src_sequence_length",shape=[None],dtype="int64")
    inputs = [src, src_sequence_length]
loader = fluid.io.DataLoader.from_generator(feed_list=inputs,capacity=10,
                                                iterable=True,use_double_buffer=True)

loader.set_batch_generator(reader,places=places)

 

之后loader会将输入的Tensor自动转化为输入需要的lodTensor

 

2.program

train_prog = fluid.Program()
startup_prog = fluid.Program()
with fluid.program_guard(train_prog, startup_prog):
    with fluid.unique_name.guard():
        pass

默认情况下网络是在default_main_program中,如果要自定义program必须使用with结构

paddle数据Tensor

原文:https://www.cnblogs.com/yangyang12138/p/12920552.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!