pytorch下的unsqueeze和squeeze的用法说明
例如unsqueeze(a,1)表示在1这个维度进行添加学习Bert模型的时候,需要使用到pytorch来进行tensor的操作,由于对pytorch和tensor不熟悉,就把pytorch中常用的、有关tensor操作的unsqueeze()、squeeze()、expand()、view()、cat()和repeat()等函数做一个总结,加深记忆。就可以用到unsqueeze(),直接上代码:也可以直接使用链式编程:tensor经过unsqueeze()处理之后,总数据量不变;维度的扩展类似于list不变直接在外面加几层[]括号。只能改变维大小为1的维,否则就会报错。返回张量的一个新视图,可以将张量的单个维度扩大为更大的尺寸。
用户评论
还不错,非常好
很好的资源,谢谢了
学习学习,感谢分享
很好的资料!
话不多说,谢谢
东西好多~~~~~
下载完还没来得及看,谢谢楼主
谢谢分享, 努力学习中
还不错 参考参考
学习中,有用的资料