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 可以自然拼接