pytorch张量的运算api

chunk

torch.chunk(tensor,chunks=2)

gather

dim为共有项          0,0    0,0    1,1    0,1

torch.reshape(a，（-1，））变一维       .reshape（（））也行（尽量这样）

torch.scatter_ 

(8条消息) 理解torch.scatter_()_lifeplayer_的博客-CSDN博客_torch.scatter

torch.split(a,2)  切2这个固定值

torch.split(a,[1,4])   切1和4

torch.squeeze       降低维度为1的维     .reshape  也行

torch.stack   拼接向量    stack是堆叠，增加维度

hstack和vstack   可以自然拼接