如何将 Numpy / PyTorch 数组的下 n 个元素设置为某个值？

Question

If I have a variable side_inputs , with .shape of [48, 161] . I want to set the last n values to some other value.

My other variable, side_input has a .shape of [45, 161] .

I want to set the last 45 elements of side_inputs to side_input with:

side_inputs[:45,:] = side_input

But I get an error:

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]

Answer 1

side_inputs[:-45, :] = side_input