Einops矩阵操作语法随记

发布时间 2023-09-07 10:13:57作者: 思念殇千寻

  参考资料:

  https://einops.rocks/1-einops-basics/

  这篇随笔主要记录使用Einops这个矩阵操作的库的时候发现的,值得记录的高频或者重要的操作:

  既然网上的教程已经很详尽了,为什么还要这样做?因为遇见一次的情况很有可能下次还会遇到,主要是写给自己看,方便以后查阅。

  1. rearrange / unsqueeze

  添加新维度

x = rearrange(ims, 'b h w c -> b 1 h w 1 c') # functionality of numpy.expand_dims
print(x.shape)
print(rearrange(x, 'b 1 h w 1 c -> b h w c').shape) # functionality of numpy.squeeze