How to select elements from a tensor according to a 2d index?

Suppose A is a 2D tensor with shape (3,4). B is a 2D tensor with shape (k,2) specify k selected index. The output is a tensor C with shape (k) described as:

C = []
for i in range(len(B)):
C = torch.tensor(C)

I wonder if there exists some Pytorch API equivalent to the above code segment?

Hi 07hyx06!


A[B[:, 0], B[:, 1]]


K. Frank