einops

[junxnone]

Reference

• einops [Code] [Docs]
• PyTorch 70.einops：优雅地操作张量维度

Brief

• rearrange
• reduce
• repeat

UseCase

UseCase	Code	Original	After
rearrange	rearrange(ims[0], 'h w c -> w h c')
...	rearrange(ims, 'b h w c -> (b h) w c')
...	rearrange(ims, 'b h w c -> (b h w c)').shape	(6, 96, 96, 3)	(165888,)
reduce	reduce(ims, 'b h w c -> h w c', 'mean')