Buffer

class torch.nn.parameter.Buffer(data=None, *, persistent=True)

一種不應被視為模型參數的 Tensor。例如，BatchNorm 的 running_mean 不是參數，而是模組狀態的一部分。

Buffers 是 Tensor 的子類別，當與 Module 一起使用時，它們具有非常特殊的屬性 - 當它們被指定為 Module 的屬性時，它們會自動添加到其 buffers 列表中，並且會出現在例如 buffers() 迭代器中。指定 Tensor 不會有這種效果。仍然可以使用 register_buffer() 函數，以顯式方式指定 Tensor。

參數

• data (Tensor) – buffer 張量。

• persistent (bool, optional) – 指示 buffer 是否為模組 state_dict 的一部分。預設值：True