From b0dd9caed4c687bae22b3979846bc961ed4bade3 Mon Sep 17 00:00:00 2001 From: wuxuejian Date: Mon, 6 Jul 2020 16:14:12 +0800 Subject: [PATCH] fix ctc label indices dim one check --- mindspore/ops/operations/nn_ops.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index eaf02efe24..fd0645acc9 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -4770,6 +4770,7 @@ class CTCLoss(PrimitiveWithInfer): def infer_shape(self, inputs, labels_indices, labels_values, sequence_length): validator.check_integer("inputs rank", len(inputs), 3, Rel.EQ, self.name) validator.check_integer("labels_indices rank", len(labels_indices), 2, Rel.EQ, self.name) + validator.check_integer("labels_indices dim one", labels_indices[1], 2, Rel.EQ, self.name) validator.check_integer("labels_values rank", len(labels_values), 1, Rel.EQ, self.name) validator.check_integer("sequence_length rank", len(sequence_length), 1, Rel.EQ, self.name) validator.check('labels_indices size', labels_indices[0], 'labels_values size',