09 2023 档案

torch和numpy的维度交换方法
摘要：Tensor的维度转置方法 ​ 在搭建神经网络的时候，经常会遇到需要交换维度的时候，比如将HWCN的Tensor维度顺序变换为NCHW顺序，此时需要用到Tensor的转置方法。 ​ 一般有以下三种方法： 1、numpy.transpose ​ 如果Tensor是由np.Array转换而来，那么可以在 阅读全文
posted @ 2023-09-10 12:13 Lumeng 阅读(670) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示