python - Slicing gives incorrect results if the values are the same

Question

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]])

It returns tensor([1, 1], dtype=torch.int32). But it should return tensor([[1, 1], [1, 1]], dtype=torch.int32). How do I make it to return this result?

question from:https://stackoverflow.com/questions/65944718/slicing-gives-incorrect-results-if-the-values-are-the-same

Answer 1

Waitting for answers