Parameter

class torch.nn.parameter.Parameter(data=None, requires_grad=True) [исходный код]

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

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

Параметры:

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