megengine.module.ConvBnRelu2d.tensors

ConvBnRelu2d.tensors(recursive=True, **kwargs)

Returns an iterable for the Tensor of the module.

参数

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

返回类型

Iterable[Parameter]