Buffer

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

一种张量，不应被视为模型参数。例如，BatchNorm的 running_mean 不是参数，但属于模块的状态。

Buffer是 Tensor 的子类，当与 Module 一起使用时，它们有一个非常特殊的属性——当它们被分配为Module的属性时，它们会自动添加到其Buffers列表中，并会出现在 buffers() 迭代器中。分配一个Tensor没有这样的效果。仍然可以通过使用 register_buffer() 函数来显式分配一个Tensor。

参数

• data (Tensor) – buffer 张量。

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