Advertisement

9-pytorch中常见的维度操作

阅读量:

1、view;reshape;Flatten:用于实现维度的整合与拆分
2、squeeze;unsqueeze:对维度进行缩减与扩展(针对维度数值为1的数据)
3、transpose;t;permute:完成维度顺序的调整(即转置操作)
4、expand;repeat:实现维度的复制与扩展

复制代码
    import torch
    
    '''
    维度变换
    1、view ;reshape;Flatten:维度合并和分解
    2、squeeze;unsqueeze:压缩维度和增加维度(相对于维度为1的数据)
    3、transpose;t;permute:维度顺顺序变换(转置)
    4、expand;repeat:维度扩展
    '''
    a = torch.rand(4, 1, 32, 32)
    
    '''
    view()的原理很简单,其实就是把原先tensor中的数据进行排列,排成一行,然后根据所给的view()中的参数从一行中按顺序选择组成最终的tensor。
    view()可以有多个参数,这取决于你想要得到的是几维的tensor,一般设置两个参数,也是神经网络中常用的(一般在全连接之前),代表二维。
    view(h,w),h代表行(想要变为几行),当不知道要变为几行,但知道要变为几列时可

全部评论 (0)

还没有任何评论哟~