摘要: torch.stack() 和 torch.cat() 都可以按照指定的维度进行拼接,但是两者也有区别,torch.satck() 是增加新的维度进行堆叠,即其维度拼接后会增加一个维度;而torch.cat() 是在原维度上进行堆叠,即其维度拼接后的维度个数和原来一致。具体说明如下: torch.s 阅读全文
posted @ 2022-07-14 22:16 jacknie23 阅读(879) 评论(0) 推荐(0) 编辑
摘要: 维度扩展 x.unsqueeze(n) 在 n 号位置添加一个维度 例子: import torch x = torch.rand(3,2) x1 = x.unsqueeze(0) # 在第一维的位置添加一个维度 x2 = x.unsqueeze(1) # 在第二维的位置添加一个维度 x3 = x. 阅读全文
posted @ 2022-07-14 21:30 jacknie23 阅读(1951) 评论(0) 推荐(0) 编辑