MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / named_parameters

Method named_parameters

python/oneflow/nn/modules/module.py:551–580  ·  view source on GitHub ↗

r""" 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. Args: prefix (str): prefix to prepend to all param

(
        self, prefix: str = "", recurse: bool = True
    )

Source from the content-addressed store, hash-verified

549 yield param
550
551 def named_parameters(
552 self, prefix: str = "", recurse: bool = True
553 ) -> Iterator[Tuple[str, Tensor]]:
554 r"""
555 named_parameters(prefix="", recurse=True) -> Iterator[Tuple[str, Tensor]]
556
557 Returns an iterator over module parameters, yielding both the
558 name of the parameter as well as the parameter itself.
559
560 Args:
561 prefix (str): prefix to prepend to all parameter names.
562 recurse (bool): if True, then yields parameters of this module
563 and all submodules. Otherwise, yields only parameters that
564 are direct members of this module.
565
566 Yields:
567 (string, Parameter): Tuple containing the name and parameter
568
569 Example::
570
571 >>> for name, param in self.named_parameters(): # doctest: +SKIP
572 ... if name in ['bias']: # doctest: +SKIP
573 ... print(param.size()) # doctest: +SKIP
574
575 """
576 gen = self._named_members(
577 lambda module: module._parameters.items(), prefix=prefix, recurse=recurse
578 )
579 for elem in gen:
580 yield elem
581
582 def buffers(self, recurse: bool = True) -> Iterator[Tensor]:
583 r"""

Callers 7

parametersMethod · 0.95
_align_tensorMethod · 0.80
initFunction · 0.80
test_remove_pruningMethod · 0.80
test_pruning_rollbackMethod · 0.80
test_rnn_pruningMethod · 0.80

Calls 2

_named_membersMethod · 0.95
itemsMethod · 0.45

Tested by 3

test_remove_pruningMethod · 0.64
test_pruning_rollbackMethod · 0.64
test_rnn_pruningMethod · 0.64