megengine.module.qat.QuantStub.tensors

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

上一页

megengine.module.qat.QuantStub.state_dict

下一页

megengine.module.qat.QuantStub.train