YouTipPytorch Torch Masked_Select
# PyTorch torch.masked_select Function
* * *
[
* * *
## Usage Example
## Example
import torch
x = torch.tensor([1,2,3,4,5])
mask = torch.tensor([True,False,True,False,True])
# Select elements based on mask
result = torch.masked_select(x, mask)
print(result)
Output result:
tensor([1, 3, 5])
* * *
[![Image 4: Pytorch torch Reference Manual]( Pytorch torch Reference Manual](