megengine.module.GroupNorm.named_tensors

GroupNorm.named_tensors(prefix=None, recursive=True, **kwargs)

Returns an iterable for key tensor pairs of the module, where key is the dotted path from this module to the tensor.

参数
  • prefix (Optional[str]) – prefix prepended to the keys.

  • recursive (bool) – if True, returns all tensors within this module, else only returns tensors that are direct attributes of this module.

返回类型

Iterable[Tuple[str, Tensor]]