viable/strict/1791916658: [Stable-ABI] Add torch::stable::masked_select. (#198048)
- PyTorch: 2264 events in the last 90 days
- PyTorch: 2234th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1791914049: [CUDA] Migrate UVM MemPool to Python Allocator (#199639)
What happened
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…
Summary assembled by rule from the sources below