Subsetting a two dimensional tensor with a one dimensional tensor

Question

I want to extract from each row of a two-dimensional tensor the column that is stored in another one dimensional tensor.

import torch
test_tensor = tensor([1,