init_on_cpu

• paddle.fluid.initializer.init_on_cpu()[源代码]

该接口设置强制变量在 cpu 上初始化。

返回：无

代码示例

import paddle.fluid as fluid
with fluid.initializer.init_on_cpu():
    step = fluid.layers.create_global_var(shape=[2,3], value=1.0, dtype='float32')