Как установить следующие n элементов массива Numpy / PyTorch в какое-то значение? - PullRequest
0 голосов
/ 11 апреля 2020

Если у меня есть переменная side_inputs, с .shape из [48, 161]. Я хочу установить последние n значения на другое значение.

Моя другая переменная, side_input имеет .shape [45, 161].

Я хочу установить последнее 45 элементов от side_inputs до side_input с:

side_inputs[:45,:] = side_input

Но я получаю ошибку:

RuntimeError: The expanded size of the tensor (48) must match the existing size (45) at non-singleton dimension 0.  Target sizes: [48, 161].  Tensor sizes: [45, 161]

1 Ответ

1 голос
/ 11 апреля 2020
side_inputs[:-45, :] = side_input
...