深度学习中框架中reshape和transpose的区别_深度学习transpose

作者：weixin_40725706 | 2024-02-27 15:39:58

踩

深度学习transpose

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录

前言
一、reshape与transpose的区别？
二、两种函数使用在哪些地方需要注意
总结

前言

今天在使用注意力机制对张量维度进行变换时，需要交换特征图通道与行列维度。因此使用reshape和transpose两个函数进行变换。
之前使用的深度学习的框架是tensorflow，这周接触到pytorch框架，发现两个框架张量（tensor）波段顺序不一样，tensorflow张量顺序为（b，w，h，c），字母代表的含义依次为batch_size，行，列，通道，而pytorch框架里张量顺序为（b，c，w，h）。而在使用不同库读取遥感影像时也会出现波段顺序不一致的情况，常用的简单tif影像读取库有PIL，scipy，cv2，tifffile，这些库通常用于rgb或tif灰度图像的读取，读取结果为（w，h，c），而对于GeoTIFF影像而言

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/154121?site