Buffer

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

一种不应被视为模型参数的 Tensor 类型。例如，BatchNorm 的 running_mean 不是参数，但却是模块状态的一部分。

Buffer 是 Tensor 子类，当与 Module 一起使用时，它们具有非常特殊的属性 – 当它们被赋值为 Module 属性时，它们会自动添加到其 buffer 列表中，并将显示在例如 buffers() 迭代器中。赋值 Tensor 不会产生这种效果。仍然可以通过使用 register_buffer() 函数显式地赋值 Tensor。

参数

• data (Tensor) – buffer 张量。

• persistent (bool, 可选) – buffer 是否是模块的 state_dict 的一部分。默认值：True