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