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())