分类：squeeze

2021
03-23

numpy的squeeze函数使用方法

reshape函数：改变数组的维数（注意不是shape大小）>>>e=np.arange(10)>>>earray([0,1,2,3,4,5,6,7,8,9])>>>e.reshape(1,1,10)array([[[0,1,2,3,4,5,6,7,8,9]]])>>>e.reshape(1,1,10)array([[[0,1,2,3,4,5,6,7,8,9]]])>>>e.reshape(1,10,1)array([[[0],[1],[2],[3],[4],[5],[6],[7],[8],[9]]])squeeze函数：从数组的形状中删除单维度条目，即把shape中为1... 继续阅读 >

2021
02-21

pytorch下的unsqueeze和squeeze的用法说明

#squeeze函数：从数组的形状中删除单维度条目，即把shape中为1的维度去掉#unsqueeze()是squeeze()的反向操作，增加一个维度，该维度维数为1，可以指定添加的维度。例如unsqueeze(a,1)表示在1这个维度进行添加importtorcha=torch.rand(2,3,1)print(torch.unsqueeze(a,2).size())#torch.Size([2,3,1,1])print(a.size())#torch.Size([2,3,1])print(a.squeeze().size())#torch.Size([2,3])print(a.squ... 继续阅读 >