Hi
My problem is :   I have a tensor with shape  output1.shape torch.Size([1, 2, 120, 89]) .
I wanted to do  something like output1.data.numpy()[0,1,:,:].   I did
auto output3=output1.squeeze()  which produced  output3.shape torch.Size([ 2, 120, 89]).
Now, how can I get [0,:,:] and [1,:,:]  separately as 2 tensors each  with shape [120,89]  ?
I can use       auto accessor=output3.accessor<float,3>();   and get the elements only by one as a 2D array.
Is there a better way to do these type of operations in C++ ?
Best Regards
JK