yes it works I think!
just a last question here,
if i want to concatenate the results based on channels, is there a faster way than:
B = input_windows_sorted[:,0,:,:,:]
for i in range(1,input_windows_sorted.size(1)):
    B=torch.cat((B,input_windows_sorted[:,i,:,:,:]),1)

for this example B will have the size of [1,8,3,3]