jack-nie-23

摘要： torch.stack() 和 torch.cat() 都可以按照指定的维度进行拼接，但是两者也有区别，torch.satck() 是增加新的维度进行堆叠，即其维度拼接后会增加一个维度；而torch.cat() 是在原维度上进行堆叠，即其维度拼接后的维度个数和原来一致。具体说明如下： torch.s 阅读全文

摘要： 维度扩展 x.unsqueeze(n) 在 n 号位置添加一个维度 例子： import torch x = torch.rand(3,2) x1 = x.unsqueeze(0) # 在第一维的位置添加一个维度 x2 = x.unsqueeze(1) # 在第二维的位置添加一个维度 x3 = x. 阅读全文