随笔分类 -  pytorch

torch中squeeze与unsqueeze用法
摘要：import torch torch中的squeeze与unsqueeze作用是去除/添加维度为1的行 例如，a=torch.randn(2,3) 那么b=a.unsqueeze(0)，b为（1,2,3）的矩阵 类似的，b=a.unsqueeze(1)，b为（2,1,3）的矩阵 b=a.unsque 阅读全文
posted @ 2019-02-14 16:42 喝姜水涂风油精 阅读(594) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示