#tensor
#тензор
Вопрос:
Форма входного тензора, как показано ниже
input =
[[ 0 0 1 2]
[ 0 3 4 5]
[ 0 6 7 8]
[ 1 9 10 11]
[ 1 12 13 14]
[ 1 15 16 17]
[ 1 18 19 20]
[ 1 21 22 23]
[ 1 24 25 26]
[ 1 27 28 29]
[ 1 30 31 32]
[ 2 33 34 35]
[ 2 36 37 38]
[ 2 39 40 41]]
И я хочу извлекать блочные элементы в соответствии с первым элементом каждой строки (например: 0,1,2), кто-нибудь поможет мне с этим, СПАСИБО!
Если есть готовая функция, было бы здорово.
Комментарии:
1. Какую библиотеку вы используете? Это может помочь авторам ответов, если вы сможете дать больше контекста вопросу. Я не уверен, какими должны быть блочные элементы.
2. Привет, так приятно получить от вас ответ, спасибо. Я использовал numpy для создания матрицы, чистый python. «Блочные» элементы здесь я имею в виду строки с одним и тем же первым элементом (я, вероятно, неправильно понял вас в вопросе), например, выше, первые три строки, первые элементы которых равны 0, а затем 8 строк, первые элементы которых имеют значение 1, и так далее. Потому что я не знаю, сколько строк в каждом «блоке», поэтому для меня это немного сложно.
3. После того, как вы сгруппировали строки, что бы вы с ними сделали? Как будут выглядеть тензоры в конечном результате?