megengine.module.qat.BatchMatMulActivation.named_tensors

BatchMatMulActivation.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.qat.BatchMatMulActivation.named_parameters

下一页

megengine.module.qat.BatchMatMulActivation.parameters