About some issue about torch.median()

torch.compile
1

when i want to use torch.median(), i found that the median of even number of data point is not as expected

torch_median
torch_median1810×872 72.7 KB

why col_medians is not tensor([2.5, 4.5, 3.5, 5.5])

the version of torch is 2.8.0+cu128.

2

From the docs:

The median is not unique for input tensors with an even number of elements in the dimension dim. In this case the lower of the two medians is returned. To compute the mean of both medians in input, use torch.quantile() with q=0.5 instead.