【pytorch】维度变换

【pytorch】维度变换

    • View操作
    • unSqueeze操作
    • 图片处理的一个案例
    • squeeze 维度删减操作
    • 维度扩展-expand
    • 维度扩展-repeat
    • 矩阵的转置操作-transpose

View操作

将一个四维的张量(b x c x h x w)转换成一个二维的张量
【pytorch】维度变换_第1张图片

对于四张图片 将每一张图像用一行向量进行表示,忽略通道信息

unSqueeze操作

增加一个维度,在指定索引位置添加维度吗,不会改变数据本身,相当于新增加一个组别
在这里插入图片描述

【pytorch】维度变换_第2张图片

图片处理的一个案例

bias相当于给每一个channel上的所有像素增加一个偏置

需要变换bias的维度
【pytorch】维度变换_第3张图片

squeeze 维度删减操作

  • b.squeeze() 删减所有维度为1

【pytorch】维度变换_第4张图片

【pytorch】维度变换_第5张图片

维度扩展-expand

应用场景:四张32通道的14 x 14大小的图片,然后添加上偏置1,32,1,1 那么需要对偏置进行扩展

【pytorch】维度变换_第6张图片

【pytorch】维度变换_第7张图片

维度扩展-repeat

这种方法会主动复制内存数据,repeat的每一个参数表示复制数据多少次

【pytorch】维度变换_第8张图片

矩阵的转置操作-transpose

【pytorch】维度变换_第9张图片

你可能感兴趣的:(人工智能,#,pytorch,pytorch,python,深度学习)