随手记录blog
import torch
list_1 = torch.tensor([1,4,5,6,0])
list_2 = [1,4,5,6,0]
TF_1 = torch.tensor([True,False,False,True,True])
TF_2 = [True,False,False,True,True]
print("tensor:")
print(list_1[TF_1])
print(list_1[TF_2])
print("list:")
# print(list_2[TF_1])
# print(list_2[TF_2])
# 下面注释的部分会报错
# 也就是只有张量才可以使用list或者tensor作为indices,张量真好用