p74368291
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2872 fix ctc label indices dim one check 

Merge pull request !2872 from wuxuejian/ctcloss_check
This commit is contained in:
mindspore-ci-bot 2020-07-06 19:13:25 +08:00 committed by Gitee
parent 3d377c51b9 b0dd9caed4
commit a6b67eb216
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ops/operations/nn_ops.py
View File

@ -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',