Tensor ‘x0_0’ ([4, 32, 256, 256]) and ‘up’  ([4, 64, 256, 256]) are concatenated together during the forward process. When I was checking the gradients after calling loss.backward(), I found the gradients assignment mismatch with the torch.cat() operation? Detailed results are showed as follows:

image867×814 396 KB