import torch as t
a=t.tensor([False,True])
b=t.tensor([[2,3,4],[1,2,3]])
print(b[0],b[1])
c=b[a]
print(c)
Output:
tensor([2, 3, 4]) tensor([1, 2, 3])
tensor([[1, 2, 3]])
import torch as t
a=t.tensor([[1,2,4],[1,2,3]])
b=t.tensor([[2,3,4],[1,2,3]])
c=a==b
print(c)
Output:
tensor([[False, False, True],
[ True, True, True]])