Identity
-
class torch.nn.Identity(*args, **kwargs)[source] -
Заполнитель оператора тождества, нечувствительный к аргументам.
- Параметры
- Форма:
-
- Вход: , где означает любое количество измерений.
- Выход: , такая же форма, как и вход.
Примеры:
>>> m = nn.Identity(54, unused_argument1=0.1, unused_argument2=False) >>> input = torch.randn(128, 20) >>> output = m(input) >>> print(output.size()) torch.Size([128, 20])
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.Identity.html