create_global_var
- paddle.static. create_global_var ( shape, value, dtype, persistable=False, force_cpu=False, name=None ) [源代码]
在全局块中创建一个新的 Tensor,Tensor 的值为 value 。
参数
shape (list[int]|tuple[int])- 指定输出 Tensor 的形状。
value (float)- 变量的值,填充新创建的变量。
dtype (str)– 初始化数据类型。
persistable (bool,可选)- 是否为永久变量,默认:False。
force_cpu (bool,可选)- 是否将该变量放入 CPU,默认值为 False。
name (str,可选) - 具体用法请参见 api_guide_Name,一般无需设置,默认值为 None。
返回
创建的 Tensor 变量。
返回
Variable。
代码示例
>>> import paddle
>>> paddle.enable_static()
>>> main_program = paddle.static.Program()
>>> startup_program = paddle.static.Program()
>>> with paddle.static.program_guard(main_program, startup_program):
... var = paddle.static.create_global_var(
... shape=[2, 3],
... value=1.0,
... dtype="float32",
... persistable=True,
... force_cpu=True,
... )
>>> var.shape
(2, 3)