快捷方式

缓冲区

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

一种不应被视为模型参数的张量类型。例如,BatchNorm 的 running_mean 不是参数,但它是模块状态的一部分。

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

参数
  • data (张量) – 缓冲区张量。

  • persistent (布尔值, 可选) – 缓冲区是否是模块的 state_dict 的一部分。默认值:True

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源