2020年11月26日
pytorch的unsqueeze与squeeze函数
摘要： torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度。 比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。 squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。 a.squeeze(dim) 就是去掉a 阅读全文
posted @ 2020-11-26 11:30 爱吃砂糖橘的白龙 阅读(242) 评论(0) 推荐(0) 编辑