MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / create_variable

Method create_variable

python/paddle/nn/layer/layers.py:1103–1154  ·  view source on GitHub ↗

Create Tensor for this layer. Parameters: name(str, optional): name of the tensor. Please refer to :ref:`api_guide_Name` . Default: None persistable(bool, optional): if set this tensor persistable. Default: False dtype(str, optional): data typ

(
        self,
        name: str | None = None,
        persistable: bool | None = None,
        dtype: DTypeLike | None = None,
    )

Source from the content-addressed store, hash-verified

1101 reason="New api in create_tensor, easier to use.",
1102 )
1103 def create_variable(
1104 self,
1105 name: str | None = None,
1106 persistable: bool | None = None,
1107 dtype: DTypeLike | None = None,
1108 ) -> Tensor:
1109 """
1110
1111 Create Tensor for this layer.
1112
1113 Parameters:
1114 name(str, optional): name of the tensor. Please refer to :ref:`api_guide_Name` . Default: None
1115
1116 persistable(bool, optional): if set this tensor persistable. Default: False
1117
1118 dtype(str, optional): data type of this parameter. If set str, it can be "bool", "float16", "float32", "float64","int8", "int16", "int32", "int64", "uint8" or "uint16". If set None, it will be "float32". Default: None
1119
1120 Returns:
1121 Tensor, created Tensor.
1122
1123 Examples:
1124 .. code-block:: pycon
1125
1126 >>> import paddle
1127
1128 >>> class MyLinear(paddle.nn.Layer):
1129 ... def __init__(self, in_features, out_features):
1130 ... super().__init__()
1131 ... self.linear = paddle.nn.Linear(10, 10)
1132 ...
1133 ... self.back_var = self.create_variable(name="linear_tmp_0", dtype=self._dtype)
1134 ...
1135 ... def forward(self, input):
1136 ... out = self.linear(input)
1137 ... paddle.assign(out, self.back_var)
1138 ...
1139 ... return out
1140
1141 """
1142 if name is not None:
1143 var_name = ".".join([self._full_name, name])
1144 else:
1145 var_name = unique_name.generate(
1146 ".".join([self._full_name, "_generated_var"])
1147 )
1148
1149 return self._helper.main_program.current_block().create_var(
1150 name=var_name,
1151 persistable=persistable,
1152 dtype=dtype,
1153 type=core.VarDesc.VarType.DENSE_TENSOR,
1154 )
1155
1156 # TODO: Add more parameter list when we need them
1157 def create_tensor(

Callers 15

select_inputFunction · 0.45
continuous_value_modelFunction · 0.45
group_normFunction · 0.45
spectral_normFunction · 0.45
_elementwise_opFunction · 0.45
array_writeFunction · 0.45
create_arrayFunction · 0.45
create_tensorFunction · 0.45
_tril_triu_opFunction · 0.45
static_forwardMethod · 0.45
_clip_by_normMethod · 0.45
shuffle_batchFunction · 0.45

Calls 4

current_blockMethod · 0.80
joinMethod · 0.45
generateMethod · 0.45
create_varMethod · 0.45

Tested by 1

run_momentum_opFunction · 0.36