I now hate shapes, reshaping, squeezing, unsqueezing
Are you using einops and einsum? I hate these somewhat less since using them. See here for more details.
Are you using einops and einsum? I hate these somewhat less since using them. See here for more details.