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.chunk(tensor,chunks=2)
dim为共有项 0,0 0,0 1,1 0,1
(8条消息) 理解torch.scatter_()_lifeplayer_的博客-CSDN博客_torch.scatter