From 19859b74f5a4ef5c09a501cd6057aa52283cec6c Mon Sep 17 00:00:00 2001 From: yepei6 Date: Tue, 2 Feb 2021 20:26:09 +0800 Subject: [PATCH] getitem pack debug --- mindspore/ops/composite/multitype_ops/_compile_utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/composite/multitype_ops/_compile_utils.py b/mindspore/ops/composite/multitype_ops/_compile_utils.py index 088e0791b8..a86e3cae37 100644 --- a/mindspore/ops/composite/multitype_ops/_compile_utils.py +++ b/mindspore/ops/composite/multitype_ops/_compile_utils.py @@ -265,8 +265,10 @@ def _generate_indices_from_tuple_of_tensor(data, tuple_index, op_name): tensor_index_shape = hyper_map(F.shape, tuple_index) broadcast_shape = const_utils.generate_broadcast_shape(tensor_index_shape, op_name) broadcast_tensors = hyper_map(F.partial(_broadcast, broadcast_shape), tuple_index) - indices = pack(broadcast_tensors) - indices = F.cast(indices, mstype.int64) + new_broadcast_tensors = () + for tensor in broadcast_tensors: + new_broadcast_tensors += (F.cast(tensor, mstype.int64),) + indices = pack(new_broadcast_tensors) return indices