init_on_cpu

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

返回:无

代码示例

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