From 3ea4469ae918af145e839d9fe68aad64ecfd8399 Mon Sep 17 00:00:00 2001 From: l30036321 Date: Wed, 13 Mar 2024 10:12:27 +0800 Subject: [PATCH] Add New Adaptation of __getitem__ --- .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py index f732a3851..3878d9518 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py @@ -698,7 +698,7 @@ def parameter_adapter(func): indices = getattr(torch._C._VariableFunctionsClass, "nonzero")(indices, as_tuple=True) return getattr(torch._C._TensorBase, "__getitem__")(input_tensor, indices) elif indices.dtype != torch.bool: - if len(indices.shape) == 1: + if not indices.shape or len(indices.shape) == 1: return func(self, input_tensor, indices.tolist()) elif len(indices.shape) == 2: result = [func(self, input_tensor, index) for index in indices.tolist()] -- Gitee