return inputs[1][:, :residual_shape[1], :residual_shape[2], :]