Hello,
lets say my network produces an output x as an 1D-tensor with [1,5,2,10]. If I slice it into two different 1D-tensors like a = x[0:1], b=x[2:3] and put them in a loss function. Will the automatic differentiation algorithm be able to do the differentiation in my network? how should the automatic differentiation algorithm know that these tensors are from my network?