Как совместить последние 2 измерения Numpy Array - PullRequest
0 голосов
/ 01 марта 2019

Я уже видел Как объединить измерения в массиве NumPy?

И попробовал это, но когда я попробовал

imgs.reshape(img.shape[:-2]+(-1))

, что я предположил, будет логичнымрасширение, я получаю сообщение об ошибке:

can only concatenate tuple (not "int") to tuple

Я собирался прокомментировать принятый ответ по связанному вопросу и напрямую обратиться к ним за помощью вместо создания новой темы для этого, но у меня неттребуется комментирование репутации.

Редактировать: Например, у меня есть входной массив, имеющий форму (x,y,8,8), где x и y могут меняться в зависимости от изображения, которое подается в функцию.Я хочу превратить его в массив формы (x,y,64).64 в этом случае никогда не меняется.

1 Ответ

0 голосов
/ 01 марта 2019

Вы должны добавить запятую, чтобы она была распознана как кортеж.Попробуйте

imgs.reshape(img.shape[:-2] + (-1,))

Или используйте распаковку:

imgs.reshape((*img.shape[:-2], -1))
...