Spec-Zone.ru › PyTorch 2

Параметр

class torch.nn.parameter.Parameter(data=None, requires_grad=True) [source]

Вид тензора, который следует рассматривать как параметр модуля.

Параметры являются подклассами Tensor, которые обладают очень особым свойством при использовании с Module — когда они присваиваются как атрибуты модуля, они автоматически добавляются в список его параметров и будут отображаться, например, в итераторе parameters(). Присвоение тензора не имеет такого эффекта. Это связано с тем, что можно захотеть кэшировать некоторое временное состояние, например, последнее скрытое состояние RNN, в модели. Если бы не было такого класса, как Parameter, эти временные данные также регистрировались бы.

Параметры
  • data (Тензор) – тензор параметра.
  • requires_grad (bool, необязательно) – требуется ли градиент для параметра. Обратите внимание, что контекст torch.no_grad() НЕ влияет на поведение по умолчанию при создании параметра — параметр по-прежнему будет иметь requires_grad=True в no_grad режиме. Подробнее см. Локальное отключение вычисления градиента. Значение по умолчанию: True

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.parameter.Parameter.html

Spec-Zone.ru

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