Spec-Zone.ru › PyTorch 2

Identity

class torch.nn.Identity(*args, **kwargs) [source]

Заполнитель оператора тождества, нечувствительный к аргументам.

Параметры
  • args (Any) – любой аргумент (не используется)
  • kwargs (Any) – любой ключевой аргумент (не используется)
Форма:
  • Вход: (∗)(*), где ∗* означает любое количество измерений.
  • Выход: (∗)(*), такая же форма, как и вход.

Примеры:

>>> 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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API