pytorch Tensor合并与切割

合并

1. cat

       

      dim表示拼接的维度。注意其他维度要保持一致。

      

2. stack

      

      dim维度前添加一个新的维度，作为原来两个tensor连接，进行区分的维度

      如果我们令 c = torch.stack([a,b], dim=1) 则：

            c[:,0,...] == a

      同样的 stack 也要求相连的维度相同。

切割

1. split

        

      split参数，1表示切分分段每一个的长度，dim是切分的维度

      所以1，dim=0，表示将0维度，每一个进行切分，切出四个结果， 每个切分结果的维度都是[1,2,5,6]

      

      当无法整除切分时，剩余的是一个切分结果。

       

      split还可以传入列表参数，表示每一个切分结果的长度，但是要求列表数之和等于原来该维度的长度。

2. chunk

     

     参数2表示总共拆分成两个部分，这里要注意如果无法整除的情况。