torch gather 3D tensor with 2D tensor as “gather map”

Question

For the given tensors of dimensions
data.shape [B,N,F] and indices.shape [N,K],
Where K are the indices of the "neighborhood&