I want to make a slice of a tensor with the same value.
import torch dist = torch.ones((10, 20), dtype=torch.int32) print(dist[[1, 2], [3, 4]])
I