[]中括号的作用(张量A[张量B])
有两种情况:
当张量B的值为int型:按照B中元素的值i,取出A中的第i行。
当张量B的值为bool型:按照B中为true的元素所在位置,取出A中对应的元素,成为一根向量。例子如下:
例1:
sim = torch.tensor([[1,2,3,4,5,6],[1,1,1,1,1,1],[2,2,2,2,2,2],[3,3,3,3,3,3],[4,4,4,4,4,4],[5,5,5,5,5,5]])
b = torch.tensor([0,2,1])
print(s