Самый простой способ индексации в измерении

У меня есть два тензора: x и y, которые имеют равные shape в первом k размеры. Второй тензор содержит индексы для извлечения значений из первого по последнее измерение. Для rank из 3, тогда вывод z должен быть таким, чтобы z[i_1, i_2,...,i_k, j] = x[i_1, i_2,...,i_k, y[i_1, i_2, ...,i_k, j]].

В настоящее время у меня есть метод, который требует изменения формы x и y, добавление индексов строк в y с использованием gather_nd и, наконец, возвращаемся к первоначальной форме. Есть ли более элегантный метод? Есть ли способ получить тензор индексов (например, np.indices), предпочтительно, не требующий знания ранга или формы, кроме того, что удовлетворять вышеуказанному условию?

4 голоса | спросил deasmhumnha 9 FebruaryEurope/MoscowbSat, 09 Feb 2019 05:31:49 +0300000000amSat, 09 Feb 2019 05:31:49 +030019 2019, 05:31:49

1 ответ


0
Нашел это!---- +: = 0 =: + ---- и ---- +: = 1 =: + ---- .
ответил deasmhumnha 9 FebruaryEurope/MoscowbSat, 09 Feb 2019 07:37:09 +0300000000amSat, 09 Feb 2019 07:37:09 +030019 2019, 07:37:09

Похожие вопросы

Популярные теги

security × 330linux × 316macos × 2827 × 268performance × 244command-line × 241sql-server × 235joomla-3.x × 222java × 189c++ × 186windows × 180cisco × 168bash × 158c# × 142gmail × 139arduino-uno × 139javascript × 134ssh × 133seo × 132mysql × 132