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

2024-01-07 22:33:32

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

常见操作：

  1. 维度变换 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)

文章来源:https://blog.csdn.net/weixin_43154149/article/details/135445663
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。