緩衝區

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

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

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

引數

• data (Tensor) – 緩衝區張量。

• persistent (bool, 可選) – 緩衝區是否是模組 state_dict 的一部分。預設值: True