Is there some way to get around this without having to avoid item assignment? Can I somehow inform that particular proxy instance that this should be okay?
You would likely want to lose the indexing on the left hand side (which makes the assignment necessarily in-place). How to do that depends on more context than shown here - as shown here, you might just keep the first column of patch_embed separately or so.