Spec-Zone.ru › PyTorch 1

Параметр

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

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

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

Параметры:
  • data (Тензор) – тензор параметра.
  • requires_grad (bool, опционально) – требуется ли градиент для параметра. Дополнительные сведения см. в разделе Локальное отключение вычисления градиента. По умолчанию: True

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

Spec-Zone.ru

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