关于torch.unsqueeze与torch.squeeze

烟中雯城

 关注

阅读 115

2022-04-21

torch内置的unsqueeze方法是增加1的维度,如下

squeeze和unsqueeze相反，是删除数据中1的维度

 

当指定数据时，删除指定位置的1的维度，当不指定参数时，删除数据中所有1的维度

也有另一种方法可以增加或删除维度，如下

 也可以指定位置

 使用这种方法如果想删除维度可以采用切片的方式将其取出，也可以使用squeeze方法。

相关推荐

PyTorch中torch.squeeze() 和torch.unsqueeze()用法

phpworkerman 55 0 0

torch.unsqueeze和torch.squeeze维度修改 #yyds干货盘点#

以前干嘛去了 61 0 0

pytorch中的torch.unsqueeze和squeeze张量维度变化问题

kmoon_b426 232 0 0

torch中 squeeze 和 unsqueeze 的不同用法

酷子腿长一米八 38 0 0

torch.squeeze() dim=1 dim=-1 dim=2

兮城 33 0 0

PyTorch的torch.cat、squeeze()、unsqueeze()和size()函数

灯火南山 221 0 0

torch.Tensor行向量转为列向量（unsqueeze）

老罗话编程 66 0 0

Pytorch维度操作-unsqueeze、squeeze、view与permute

M4Y 79 0 0

pytorch中squeeze()和unsqueeze()函数

霍华德 46 0 0

torch.jit.trace与torch.jit.script

海滨公园 44 0 0

精彩评论（0）

0 0 举报