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,
)
| 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( |