oneflow.nn.Module.named_parameters¶
-
Module.
named_parameters
(prefix='', recurse=True) → Iterator[Tuple[str, Tensor]]¶ Returns an iterator over module parameters, yielding both the name of the parameter as well as the parameter itself.
- Parameters
prefix (str) – prefix to prepend to all parameter names.
recurse (bool) – if True, then yields parameters of this module and all submodules. Otherwise, yields only parameters that are direct members of this module.
- Yields
(string, Parameter) – Tuple containing the name and parameter
Example:
>>> for name, param in self.named_parameters(): ... if name in ['bias']: ... print(param.size())