viable/strict/1791916658: [Stable-ABI] Add torch::stable::masked_select. (#198048)

发生了什么

Exposes masked_select through the stable ABI as torch::stable::masked_select . torchvision's stable-ABI currently utilizes a local masked_select helper . Summary Adds torch::stable::masked_select , which calls aten::masked_select through the dispatcher and returns a 1-D tensor of the elements of self where mask is true. The wrapper has the same shape as torch::stable::index_select . Adds a my_masked_select op to the…

摘要按规则整理自下方来源原文

来源