megengine.module.Module.buffers

Module.buffers(recursive=True, **kwargs)[源代码]

返回该模块中对于buffers的一个可迭代对象。

Buffer被定义为是 Tensor 且不是 Parameter

参数

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

返回类型

Iterable[Tensor]