Выбор по разному индексу столбца для каждой строки в тензоре ⇐ Python

[Anonymous]

У меня есть тензор Pytorch

Код: Выделить всё

t = torch.tensor(
[[1.0, 1.5, 0.5, 2.0],
[5.0, 3.0, 4.5, 5.5],
[0.5, 1.0, 3.0, 2.0]]
)

Код: Выделить всё

t[:, [-1]]

выдает мне значение последнего столбца каждой строки:

Код: Выделить всё

tensor([[2.0000],
[5.5000],
[2.0000]])

Однако я хочу разделить значения в разных столбцах на строку. Например, в t для 1-й, 2-й и 3-й строки я хочу разрезать индексы 2, -1, 0 соответственно, чтобы получить следующий тензор:

Код: Выделить всё

tensor([[0.5],
[5.5],
[0.5]])

Как это сделать в Torch?

Подробнее здесь: https://stackoverflow.com/questions/729 ... -in-tensor