Как объединить строки матрицы Pytorch с одним и тем же элементом? ⇐ Python

[Anonymous]

Например, входная матрица N*2 = torch.tensor([[1, 2], [2, 3],[4, 5]]) ,
ожидаемая выходная матрица равна torch.tensor([[1, 3],[4, 5]]), поскольку первая и вторая строки имеют один и тот же элемент 2.
Как это может быть достигнуто? Заранее спасибо.

Подробнее здесь: https://stackoverflow.com/questions/781 ... me-element