Change size of Tensor

Hi everyone,
I have a simple question,
If I have a tensor with a shape of [1,3,320,320] and I want to convert him to [1,2,320,320]
what is the simple way to use?


Because the second tensor has less values contained, u cannot change the shape without losing some of the values in the first tensor. Let’s say the shape of A is (1, 3, 320, 320), the desired shape is (1, 2, 320, 320), a simple way to do so is by slicing:

START, END = 0, 2
b = A[ : , START:END, : , : ]