Buffer

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

一種 Tensor，但不應被視為模型引數。例如，BatchNorm 的 running_mean 不是引數，但屬於模組的狀態。

Buffers 是 Tensor 的子類，當它們與 Module 一起使用時，具有一個非常特殊的屬性——當它們被指定為 Module 的屬性時，它們會被自動新增到其 buffer 列表中，並會出現在例如 buffers() 迭代器中。指定一個 Tensor 不會產生這樣的效果。仍然可以透過使用 register_buffer() 函式顯式地指定一個 Tensor。

引數

• data (Tensor) – buffer tensor。

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