force_init_on_cpu

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

该接口获得一个是否强制在CPU上初始化变量的布尔型标志位。

返回:状态,是否应强制在CPU上强制进行变量初始化

返回类型:bool

代码示例

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