I have two net works A and B, the output is out = A(x1) + B(x2), it is a matrix. Now I want use out.backward(matrixj). I know there should be a maxtrxj, which has the same size with out, but I’m not sure if it is ok I just use a matrix with all element is 1.