缓冲区 ¶

类 torch.nn.parameter.Buffer(data=None, *, persistent=True)[source][source] ¶

一种不应被视为模型参数的张量。例如，BatchNorm 的 running_mean 不是一个参数，而是模块状态的一部分。

缓冲区是 Tensor 子类，当与 Module 一起使用时具有非常特殊的属性 - 当它们被分配为模块属性时，它们会自动添加到其缓冲区列表中，例如在 buffers() 迭代器中会出现。分配张量没有这种效果。仍然可以通过使用 register_buffer() 函数显式地将张量分配。