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

上一页

megengine.module.GroupNorm.named_parameters

下一页

megengine.module.GroupNorm.parameters