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