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.
Best regards
Thomas