megengine.traced_module.TracedModule.parameters

TracedModule.parameters(recursive=True, **kwargs)

Returns an iterable for the Parameter of the module.

参数

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

返回类型

Iterable[Parameter]

上一页

megengine.traced_module.TracedModule.named_tensors

下一页

megengine.traced_module.TracedModule.register_forward_hook