例如unsqueeze(a,1)表示在1这个维度进行添加学习Bert模型的时候，需要使用到pytorch来进行tensor的操作，由于对pytorch和tensor不熟悉，就把pytorch中常用的、有关tensor操作的unsqueeze()、squeeze()、expand()、view()、cat()和repeat()等函数做一个总结，加深记忆。就可以用到unsqueeze()，直接上代码：也可以直接使用链式编程：tensor经过unsqueeze()处理之后，总数据量不变；维度的扩展类似于list不变直接在外面加几层[]括号。只能改变维大小为1的维，否则就会报错。返回张量的一个新视图，可以将张量的单个维度扩大为更大的尺寸。
用户评论