Einops矩阵操作语法随记

原创于 2023-09-07 10:10:00 发布 · 161 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#矩阵 #线性代数

本文记录在使用Einops库时遇到的重要操作，如rearrange（类似numpy的expand_dims用于添加新维度）和unsqueeze，以及squeeze的功能，目的是为日后查询提供便利。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

　　参考资料：

　　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