(beta)torch_npu.npu.set_mm_bmm_format_nd 接口原型torch_npu.npu.set_mm_bmm_format_nd(bool) 功能描述设置线性module里面的mm和bmm算子是否用ND格式。 调用示例>>>torch_npu.npu.set_mm_bmm_format_nd(True) 父主题: torch_npu