torch_npu.npu.get_mm_bmm_format_nd
torch_npu.npu.get_mm_bmm_format_nd()
确认线性module里面的mm和bmm算子是否有使能ND格式,如果使能了ND,返回True,否则,返回False。
参数解释:
无参数。
返回值:
bool型。
约束条件:
无
示例:
>>>torch_npu.npu.set_mm_bmm_format_nd() True
父主题:
torch_npu