pytorch API (1)
unsqueeze 增加维度
squeeze减少维度
squeeze翻译 压缩
permute() 根据括号中的位置进行处理,进行位置的调整。
view()cat就是直接拼接起来,后面的数字代表是在 横向还是纵向
stack 会增加新的维度,后面数字 即为代表是在哪个维度上进行拼接。。
https://blog.****.net/david0611/article/details/81058197
在一个维度上进行分块 chunks参数是分几个 split_size 参数是分块的长度大小
input 原始张量,dim是指在哪个维度上进行索引,index即为在dim所指的维度上进行相应的选取 然后填入到新的tensor中,新的tensor 和原始的input 维度都是一样的 然后假设在dim=1 上进行处理,其他维度该怎么放置就怎么放置,只是在dim=1的维度上进行索引变化处理。
和gather 类似,选出再某个dim上的相应的数据
参考
https://blog.****.net/gyt15663668337/article/details/91345951