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