oneflow.nn.Module.buffers

Module.buffers(recurse=True)Iterator[Tensor]

Returns an iterator over module buffers.

Parameters

recurse (bool) – if True, then yields buffers of this module and all submodules. Otherwise, yields only buffers that are direct members of this module.

Yields

oneflow.Tensor – module buffer

Example:

>>> for buf in model.buffers(): 
...     print(type(buf), buf.size()) 
<class 'oneflow.Tensor'> oneflow.Size([10])