ReplicationPad1d
-
class torch.nn.ReplicationPad1d(padding)[source] -
Заполняет тензор входными границами, повторяя их.
Для заполнения в
N-мерном случае используйтеtorch.nn.functional.pad().- Параметры
-
padding (int, tuple) – размер заполнения. Если это
int, используется одинаковое заполнение для всех границ. Если это 2-tuple, используется (, )
- Форма:
-
- Вход: или .
-
Вывод: или , где
Примеры:
>>> m = nn.ReplicationPad1d(2) >>> input = torch.arange(8, dtype=torch.float).reshape(1, 2, 4) >>> input tensor([[[0., 1., 2., 3.], [4., 5., 6., 7.]]]) >>> m(input) tensor([[[0., 0., 0., 1., 2., 3., 3., 3.], [4., 4., 4., 5., 6., 7., 7., 7.]]]) >>> # using different paddings for different sides >>> m = nn.ReplicationPad1d((3, 1)) >>> m(input) tensor([[[0., 0., 0., 0., 1., 2., 3., 3.], [4., 4., 4., 4., 5., 6., 7., 7.]]])
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.ReplicationPad1d.html