Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
batch_norm shape 为2d时，format从ND变为了NCHW
Ascend950 CANN9.0
构造输入为2d，format为nchw的用例
构造输入为2d，format为nchw的用例，执行成功
构造输入为2d，format为nchw的用例，执行失败
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
batch_norm shape 为2d时，format从ND变为了NCHW
Environment / 环境信息 (Mandatory / 必填)
Ascend950
CANN9.0
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
构造输入为2d，format为nchw的用例
Describe the expected behavior / 预期结果 (Mandatory / 必填)
构造输入为2d，format为nchw的用例，执行成功
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
构造输入为2d，format为nchw的用例，执行失败
Special notes for this issue/备注 (Optional / 选填)