2022年3月17日
pytorch中的squeeze函数、unsqueeze函数
摘要： 1、 squeeze()函数 去除size为1的维度，包括行和列 至于维度大于等于2时，squeeze()不起作用 torch.unsqueeze(A，N)函数的作用减少数组A指定位置N的维度，如果指定的维度大于1，那么将操作无效，如果不指定维度N，那么将删除所有维度为1的维度 行变的例子 >>> 阅读全文
posted @ 2022-03-17 09:16 时光如你般美好 阅读(705) 评论(0) 推荐(0) 编辑
、