缓冲区

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