megengine.module.qat.QATModule.named_buffers

QATModule.named_buffers(prefix=None, recursive=True, **kwargs)

返回可遍历模块中 key 与 buffer 的键值对的可迭代对象，其中 key 为从该模块至 buffer 的点路径（dotted path）。

Buffer被定义为是 Tensor 且不是 Parameter

参数

  • prefix (Optional[str]) – 加在每个键（key）前的前缀。

  • recursive (bool) – 如果为 True ，则返回所有当前模块中的buffer，否则只返回属于该模块的直接属性。

  • prefix – Optional[str]:

返回类型

Iterable[Tuple[str, Tensor]]

上一页

megengine.module.qat.QATModule.modules

下一页

megengine.module.qat.QATModule.named_children