Describe the current behavior / 问题描述 (Mandatory / 必填)

xai api样例代码 mindspore_xai.benchmark.ClassSensitivity 在ascend执行报错

Environment / 环境信息 (Mandatory / 必填)

• Hardware Environment(Ascend/GPU/CPU) / 硬件环境:

Please delete the backend not involved / 请删除不涉及的后端:
/device ascend

• Software Environment / 软件环境 (Mandatory / 必填):
  -- MindSpore version (e.g., 1.7.0.Bxxx) :mindspore 2.0.0.20230130 [sha1]:9aaa6730 xai 2.0.0.20230130
  -- Python version (e.g., Python 3.7.5) :3.9.11
  -- OS platform and distribution (e.g., Linux Ubuntu 16.04):
  -- GCC/Compiler version (if compiled from source):

• Excute Mode / 执行模式 (Mandatory / 必填)(PyNative/Graph):

Please delete the mode not involved / 请删除不涉及的模式:
/mode pynative

Related testcase / 关联用例 (Mandatory / 必填)

test_doctest_mindspore_xai_benchmark_attribution_class_sensitivity

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

1. 复制mindspore_xai.benchmark.ClassSensitivity 样例代码
2. 样例代码中的LeNet5 在models仓里倒入
3. 在910环境运行样例代码

Describe the expected behavior / 预期结果 (Mandatory / 必填)

样例代码正常执行，无报错

Related log / screenshot / 日志 / 截图 (Mandatory / 必填)

Special notes for this issue/备注 (Optional / 选填)