paddlepaddle关于使用dropout小案例

import numpy as np
import paddle.fluid as fluid
import paddle.fluid.layers as layers

debug = True
bs = 1
c = 1
h, w = 3, 1

main_program = fluid.Program()
startup_program = fluid.Program()
with fluid.program_guard(main_program, startup_program):
    x = fluid.layers.data(name='x', shape=[-1, c, w, h], dtype='float32')
    # hidden = fluid.layers.fc(input=x, size=32, act='relu')
    hidden = fluid.layers.dropout(x, dropout_prob=0.5)
    # loss = fluid.layers.cross_entropy(
    #     input=fluid.layers.fc(hidden, size=10, act='softmax'),
    #     label=fluid.layers.data(name='label', shape=[1], dtype='int64'))

if debug:
    test_program = main_program.clone(for_test=True)
else:
    test_program = main_program.clone()

place = fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(startup_program)
x_data = np.random.rand(bs, c, w, h).astype(np.float32)
print(exe.run(main_program, fetch_list=[x,hidden], feed={'x': x_data}))

print(x_data)
print(exe.run(test_program, fetch_list=[x,hidden], feed={'x': x_data}))

结果：

[array([[[[0.5775851 , 0.78441525, 0.4060972 ]]]], dtype=float32), array([[[[0.        , 0.78441525, 0.        ]]]], dtype=float32)]
[[[[0.5775851  0.78441525 0.4060972 ]]]]
[array([[[[0.5775851 , 0.78441525, 0.4060972 ]]]], dtype=float32), array([[[[0.28879255, 0.39220762, 0.2030486 ]]]], dtype=float32)]