[Slice] Support multi-dim bool tensor index #74193
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR Category
Execute Infrastructure
PR Types
Bug fixes
Description
pcard-67164
getitem 无法正确处理含有 秩>2 的布尔高级索引。
以 Tensor([3,4,5],"float32"), (slice(None,None,None),tuple(Tensor([4,5],"bool")) 为例,
parseIndex 无法识别出布尔索引会 consume input 的 1 和 2 两个维度,而是按照整数索引的处理方式认为其只作用在第 1 维度上,从而导致在 dealWithAdvancedIndex 拿到的转置的 tensor 的形状为 [4, 3, 5],而不是预期的 [4, 5, 3]。最终引发了维度不匹配报错: