tensor笔记7--tensor的各种重载

a = torch.Tensor([[1, 2, 3, 7, 8, 5, 4, 9],[1, 3, 4, 6, 1, 3, 3, 9]])
print(a[0])
b = torch.Tensor([1, 1])
c = torch.Tensor([1, 2])
for i in c:index = i == bprint(index)print(a[index])

输入： 

tensor([1., 2., 3., 7., 8., 5., 4., 9.])
tensor([True, True])
tensor([[1., 2., 3., 7., 8., 5., 4., 9.],[1., 3., 4., 6., 1., 3., 3., 9.]])
tensor([False, False])
tensor([], size=(0, 8))