Some operations are quite tensor-ish. For instance the re-shaping which einops writes like this Rearrange('b c h w -> b (c h w)') is literally a tensor pr

Some operations are quite tensor-ish. For instance the re-shaping which einops writes like this Rearrange('b c h w -> b (c h w)') is literally a tensor product of the vector spaces indexed by c,h,w. The same space that np.kron maps into (possibly modulo the order of indices) rather than np.hstack (of row vectors) which maps to the direct sum, ⊕ not ⊗.