You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

6 lines
68 B
Python

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