【代码学习】einops，更简单的张量变化

算法研究员

已于 2024-01-07 22:04:57 修改

阅读量514

点赞数 7

分类专栏：【杂学】文章标签：学习

于 2024-01-07 21:52:10 首次发布

本文链接：https://blog.csdn.net/weixin_43154149/article/details/135445663

版权

【杂学】专栏收录该内容

23 篇文章 0 订阅

订阅专栏

本文介绍了在GitHub上的官方Einops教程中关于如何使用Einops库进行维度变换，包括rearrange（重新排列）、flatten（展平）以及空间到深度和深度到空间转换的示例。

摘要由CSDN通过智能技术生成

官方教程：https://github.com/arogozhnikov/einops/blob/master/docs/2-einops-for-deep-learning.ipynb

常见操作：

维度变换 rearrange

维度变换
y = rearrange(x, 'b c h w -> b h w c') # 已经 表明 x的每个轴 变量 `b c h w`
guess(y.shape)

flatten
y = rearrange(x, 'b c h w -> b (c h w)')
guess(y.shape)

space-to-depth
y = rearrange(x, 'b c (h h1) (w w1) -> b (h1 w1 c) h w', h1=2, w1=2)
guess(y.shape)

depth-to-space
y = rearrange(x, 'b (h1 w1 c) h w -> b c (h h1) (w w1)', h1=2, w1=2)
guess(y.shape)

关注博主即可阅读全文