oneflow.nn.Module.named_buffers

Module.named_buffers(prefix='', recurse=True)Iterator[Tuple[str, Tensor]]

Returns an iterator over module buffers, yielding both the name of the buffer as well as the buffer itself.

Parameters
  • prefix (str) – prefix to prepend to all buffer names.

  • 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

(string, oneflow.Tensor) – Tuple containing the name and buffer

Example:

>>> for name, buf in self.named_buffers(): 
...    if name in ['running_var']: 
...        print(buf.size())