#502 masked_scatter

Merged
Erpim merged 10 commits from lzh/MSAdapter:0510 into master 11 months ago
lzh commented 1 year ago
zoulq changed title from masked_scatter to [WIP]masked_scatter 1 year ago
zoulq commented 1 year ago
Collaborator
依赖0416后masked_scatter接口功能
lzh changed title from [WIP]masked_scatter to masked_scatter 11 months ago
Erpim commented 11 months ago
Collaborator
/lgtm
zoulq reviewed 11 months ago
msadapter/pytorch/tensor.py
@@ -3731,0 +3732,4 @@
input = cast_to_ms_tensor(self)
mask = cast_to_ms_tensor(mask).bool()
tensor = cast_to_ms_tensor(tensor)
output = input.masked_scatter(mask, tensor)
zoulq commented 11 months ago
ms.tensor.masked_scatter接口也显示只能支持cpu平台,实测其他平台是否能支持? 如果不能支持用其他接口拼接是否能支持?
lzh commented 11 months ago
Poster
ascend上支持,已验
lzh commented 11 months ago
Poster
GPU上不支持,已更新supportedlist、跳过相关用例
Erpim commented 11 months ago
Collaborator
MindSpore:MindSpore此API实现功能与PyTorch基本一致。但是PyTorch支持 mask 与Tensor本身的双向广播, MindSpore只支持 mask 广播到Tensor本身。
Erpim closed this pull request 11 months ago
lzh reopened this pull request 11 months ago
zoulq reviewed 11 months ago
ConstraintList.md
@@ -157,6 +157,7 @@
| Tensor.repeat_interleave | 暂不支持`output_size`入参 |
| Tensor.index_reduce | 暂不支持`reduce`="mean" |
| Tensor.index_reduce_ | 暂不支持`reduce`="mean"和图模式 |
| Tensor.masked_scatter | 暂不支持`input`广播到`mask` |
zoulq commented 11 months ago
加一下GPU限制
Erpim merged commit 3615ccf37c into master 11 months ago
The pull request has been merged as 3615ccf37c.
Sign in to join this conversation.
No reviewers
No Label
No Milestone
No Assignees
3 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Loading…
There is no content yet.