缓冲区

类torch.nn.parameter.Buffer(data=None, *, persistent=True)[源代码]

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

Buffers 是 Tensor 的子类，在与 Module 一起使用时具有特殊属性：当它们被赋值为 Module 属性时，会自动添加到该模块的 buffers 列表中，并在例如 buffers() 迭代器中显示。直接将 Tensor 赋值给属性不会产生这样的效果。但是，仍然可以通过使用 register_buffer() 函数显式地将 Tensor 注册为 buffer。

参数

• data (Tensor) – 张量缓冲区。

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