Buffer

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

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

Буферы являются подклассами Tensor и обладают особым свойством при использовании с Module: при назначении в качестве атрибутов Module они автоматически добавляются в список буферов модуля и отображаются, например, в итераторе buffers(). Присваивание Tensor не приводит к такому результату. Тем не менее Tensor можно зарегистрировать в качестве буфера с помощью функции register_buffer().

Параметры:

• data (Tensor) – тензор буфера.
• persistent (bool, необязательно) – указывает, входит ли буфер в state_dict модуля. По умолчанию: True