Skip to content

Commit ec1b290

Browse files
add padding_idx setter (#74946)
1 parent 356087d commit ec1b290

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

python/paddle/nn/layer/common.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1870,6 +1870,10 @@ def __init__(
18701870
def padding_idx(self):
18711871
return self._padding_idx
18721872

1873+
@padding_idx.setter
1874+
def padding_idx(self, value):
1875+
self._padding_idx = value
1876+
18731877
@param_one_alias(["x", "input"])
18741878
def forward(self, x: Tensor) -> Tensor:
18751879
return F.embedding(

test/legacy_test/test_nn_dtype_device_bias.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -508,6 +508,9 @@ def test_padding_idx(self):
508508
layer = self.api(32, 16, padding_idx=2)
509509
assert layer._padding_idx == layer.padding_idx
510510

511+
layer.padding_idx = 5
512+
assert layer._padding_idx == 5
513+
511514

512515
if __name__ == '__main__':
513516
unittest.main()

0 commit comments

Comments
 (0)