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