megengine.traced_module.TracedModule.buffers

TracedModule.buffers(recursive=True, **kwargs)

Returns an iterable for the buffers of the module.

Buffer is defined to be Tensor excluding Parameter.

参数

recursive (bool) – if True, returns all buffers within this module, else only returns buffers that are direct attributes

返回类型

Iterable[Tensor]

上一页

megengine.traced_module.TracedModule.apply

下一页

megengine.traced_module.TracedModule.children