# %%
import torch
a=torch.tensor([1,2,3])
a[[True,False,True]]
# %%